使用device = torch.device('cuda:0')之前需要先用os.environ["CUDA_VISIBLE_DEVICES"] = '0'吗
时间: 2023-05-21 21:07:37 浏览: 207
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
回答:不一定,如果你想使用GPU 0作为计算设备,那么需要先设置环境变量CUDA_VISIBLE_DEVICES,然后再使用device = torch.device('cuda:0')来将计算设备设置为GPU 0。如果不需要使用GPU 0或者CUDA_VISIBLE_DEVICES已经设置好了,就可以直接使用device = torch.device('cuda:0')。
阅读全文