pytorch 指定GPU时,可能出现的问题

一般我们指定GPU训练时,用到的是:
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"

今天遇到一个有趣的问题,在我利用以上语句进行指定GPU的时候,上面语句失效,及无论我如何进行指定,都无法实现仅利用‘1,2’块GPU操作,总是利用到原来的0块GPU。
**解决方法:于是我将os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"代码放到了import os后面解决了!!!!!
即:如下

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"

语句不在失效~!!!!!!不知道什么原因。

你可能感兴趣的:(安装编译的问题,pytorch,深度学习)