问题: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块儿