pytorch 指定GPU时,可能出现的问题
一般我们指定GPU训练时,用到的是:os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"今天遇到一个有趣的问题,在我利用以上语句进行指定GPU的时候,上面语句失效,及无论我如何进行指定,都无法实现仅利用‘1,2’块GPU操作,总是利用到原来的0块GPU。**解决方法:于是我将os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"代...
原创
2019-12-10 11:23:03 ·
1242 阅读 ·
4 评论