CUDA GPU 隔离 CUDA_VISIBLE

场景：

有一台服务器，服务器上有多块儿GPU可以供使用，但此时只希望使用第2块和第4块GPU，但是我们希望代码能看到的仍然是有两块GPU，分别编号为0,1，这个时候我们可以使用环境变量CUDA_VISIBLE_DEVICES来解决这个问题。
比如：

CUDA_VISIBLE_DEVICES=1  只有编号为1的GPU对程序是可见的，在代码中gpu[0]指的就是这块儿GPU
CUDA_VISIBLE_DEVICES=0,2,3  只有编号为0,2,3的GPU对程序是可见的，在代码中gpu[0]指的是第0块儿，gpu[1]指的是第2块儿，gpu[2]指的是第3块儿
CUDA_VISIBLE_DEVICES=2,0,3  只有编号为0,2,3的GPU对程序是可见的，但是在代码中gpu[0]指的是第2块儿，gpu[1]指的是第0块儿，gpu[2]指的是第3块儿

如何设置环境变量?

临时设置:

Linux： export CUDA_VISIBLE_DEVICES=1
windows:  set CUDA_VISIBLE_DEVICES=1

永久设置:

linux:
在~/.bashrc 的最后加上export CUDA_VISIBLE_DEVICES=1，然后source ~/.bashrc
windows:
打开我的电脑环境变量设置的地方，直接添加就行了。

CUDA GPU 隔离 CUDA_VISIBLE_DEVICES相关推荐

目的：使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU
目的:使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU 目的:使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GP ...
qt程序中使用环境变量_目的：使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU...
目的:使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU设备环境: 系统:Win10 环境:Opencv3.4.0/4.1.0 显卡:1080ti 问题 ...
目的：使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU设备
目的:使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU设备环境: 系统:Win10 环境:Opencv3.4.0/4.1.0 显卡:1080ti 问题 ...
适用于CUDA GPU的Numba例子
• 适用于CUDA GPU的Numba例子矩阵乘法这是使用CUDA内核的矩阵乘法的简单实现: @cuda.jit def matmul(A, B, C): """Pe ...
CUDA: GPU高性能运算
CUDA: GPU高性能运算 2013-10-11 22:23 5650人阅读评论(0) 收藏举报分类: CUDA(106) 目录(?)[+] 0 序言 CUDA是异构编程的一个大头,洋洋洒洒的 ...
CUDA GPU编程
文章目录 CUDA GPU并行化的工作流程: 修饰符 `__global__` `__host__` `__device__` 内存分配 CUDA 随着显卡的发展,GPU越来越强大,而且GPU为显示图 ...
CPU与CUDA(GPU)的计算能力对比之二: Keras Resnet 运算效率比较
CPU与CUDA(GPU)的计算能力对比之二: Keras Resnet 运算效率比较结论: CUDA(GPU : NVIDIA RTX2070 MQ 笔记本版本) 启动后,效率将近是 CPU 单独 ...
torch+cuda gpu并行计算
torch+ CUDA gpu 并行计算环境配置需要注意 GPU算力值 cuda版本显卡驱动版本 torch版本之间的匹配 torch版本可以根据cuda版本来先根据驱动版本和显卡算力(显卡算 ...
Python+PyCharm+PyTorch+Cuda/GPU 安装步骤
Python+PyCharm+PyTorch+Cuda/GPU 安装步骤一. 无nvidia显卡,直接用CPU跑程序时二.有Nvidia显卡,想用GPU跑程序时 2.1 先查版本! 2.2 下载安 ...

CUDA GPU 隔离 CUDA_VISIBLE_DEVICES

场景：

如何设置环境变量?

CUDA GPU 隔离 CUDA_VISIBLE_DEVICES相关推荐

最新文章

热门文章