os.environ[“CUDA_VISIBLE_DEVICES“]=‘1‘无效?

os.environ["CUDA_VISIBLE_DEVICES"] = '1'
指定代码运行在第1块显卡,但是该设置无效,程序仍运行在第0块显卡。

解决方案:
将该代码放到import torch之前

你可能感兴趣的:(深度学习)