使用os.environ[‘CUDA_VISIBLE_DEVICES‘]指定GPU无法生效解决方案

在使用pytorch的时候利用下面的语句指定GPU为仅为"6"，但是用nvidia-smi查看GPu使用时，仍默认为"0"号

import pytorch
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '6'

解决方案：
将上述语句放到 import pytorch 之前

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '6'
import pytorch