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

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

文章目录

  • os.environ['CUDA_VISIBLE_DEVICES'] = 单值作用
  • 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。

你可能感兴趣的:(深度学习,人工智能)