tensorflow: CUDA_VISIBLE_DEVICES
场景:有一台服务器,服务器上有多块儿GPU可以供使用,但此时只希望使用第2块和第4块GPU,但是我们希望代码能看到的仍然是有两块GPU,分别编号为0,1,这个时候我们可以使用环境变量CUDA_VISIBLE_DEVICES来解决这个问题。比如:CUDA_VISIBLE_DEVICES=1只有编号为1的GPU对程序是可见的,在代码中gpu[0]指的就是这块儿GPUCUDA_VISIB...
转载
2019-04-04 18:22:24 ·
4573 阅读 ·
0 评论