【引路帖】【使用GPU】使用os.environ[‘CUDA_VISIBLE_DEVICES‘]指定GPU无法生效解决方案
问题:os.environ[‘CUDA_VISIBLE_DEVICES‘]无法生效参考链接:详解tip:当只使用一块GPU的时候,不管你设置的是服务器上的几号GPU,在代码运行中,都是当做GPU0当使用多块GPU的时候,不管你设置的使用服务器上的哪几块GPU,在代码运行中,都是按照GPU0,GPU1…进行编号例如:CUDA_VISIBLE_DEVICES=2,0,3 只有编号为0,2,3的GPU对程序是可见的,但是在代码中gpu[0]指的是第2块儿,gpu[1]指的是第0块儿,gpu[2]指的
原创
2021-01-10 11:33:19 ·
6436 阅读 ·
0 评论