os.environ[‘CUDA_VISIBLE_DEVICES‘] = 的用法

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

os.environ[‘CUDA_VISIBLE_DEVICES’] = 单值作用

一定要在import torch之前定义这句话

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1' 

---

它的意思是将第二张显卡设置为环境仅可见，就是说用到gpu的地放都会默认使用第二长显卡来跑

但是一定有一点特别需要注意

device = 'cuda:1'

这种类似的代码就不能这样写了，因为我们的os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘1’ 这行代码不但将第二张显卡设置成了仅可见，还将该显卡的序列号该成了0，也就是说，现在需要使用

device = 'cuda:0’
#或者
device = 'cuda’

才能使用到第二张gpu，不然的话代码是会报错的

os.environ[‘CUDA_VISIBLE_DEVICES’] = 多值的作用

比如

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '3,4'

它的意思就变成了 只把第一张gpu和第二长gpu改成当前系统可见gpu，并且把gpu的序列值第一张改成3，第二张改成4。