CUDA GPU 隔离 CUDA_VISIBLE_DEVICES

最新推荐文章于 2025-06-04 16:42:15 发布

__Lingyue__

最新推荐文章于 2025-06-04 16:42:15 发布

阅读量2.5w

点赞数 7

CC 4.0 BY-SA版权

分类专栏：杂记文章标签： cuda GPU cuda_visible_devices tensorflow 机器学习

本文链接：https://blog.youkuaiyun.com/cfarmerreally/article/details/80321276

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

场景：

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