一般我们指定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"
语句不在失效~!!!!!!不知道什么原因。