设置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()
聚焦前沿AI与大模型技术探索，汇聚开发者及爱好者，共享开源项目、学习资源与行业资讯。
更多推荐
14
0- 0
已为社区贡献1条内容
所有评论(0)