设置os.environ[‘CUDA_VISIBLE_DEVICES’]=‘1‘ 不生效
今天在写pytorch代码时,设置os.environ['CUDA_VISIBLE_DEVICES']=‘1‘
不生效。最后发现原因是,这一行代码放置的位置不对。
以下是错误代码:
GpuNum = torch.cuda.device_count()
os.environ['CUDA_VISIBLE_DEVICES']=‘1‘
解决方案:
一定要把os.environ['CUDA_VISIBLE_DEVICES']=‘1‘
放在所有访问GPU的代码之前。
os.environ['CUDA_VISIBLE_DEVICES']=‘1‘
GpuNum = torch.cuda.device_count()