设置os.environ['CUDA_VISIBLE_DEVICES']=‘1‘ 不生效

设置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()

你可能感兴趣的:(技术类)