【引路帖】【使用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]指的是第3块儿

参考链接:tensorflow中使用指定的GPU及GPU显存 CUDA_VISIBLE_DEVICES

你可能感兴趣的:(Ubuntu,Python,cuda)