Pytorch 单机多卡设定GPU os.environ[‘CUDA_VISIBLE_DEVICES‘] 仍旧在GPU0上运行的错误

问题描述：

        在深度学习代码中添加了以下代码指定运行的GPU

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

但是使用

torch.cuda.current_device()

查看发现仍旧在GPU0上运行。

原因：

        在代码中调用的其他文件中也有

import torch

此操作会清除os.environ的效果。

解决方法：

        将以下代码放在.py文件的最开头

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'