CUDA GPU 隔离 CUDA_VISIBLE_DEVICES

CUDA GPU 隔离：使用CUDA_VISIBLE_DEVICES

最新推荐文章于 2025-10-22 13:51:38 发布

原创

最新推荐文章于 2025-10-22 13:51:38 发布 · 2.5w 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#cuda #GPU #cuda_visible_devices #tensorflow #机器学习

在多GPU服务器上，为确保代码仅使用第2和第4块GPU，可设置环境变量CUDA_VISIBLE_DEVICES。通过此变量，可将GPU编号重置为0,1，临时设置方式为export CUDA_VISIBLE_DEVICES=2,4，永久设置涉及系统配置。" 132093382,8753399,JavaScript实现近邻传播算法详解,"['JavaScript', '算法', '开发语言', '数据处理']

场景：

有一台服务器，服务器上有多块儿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块儿