CUDA GPU 隔离 CUDA_VISIBLE_DEVICES

场景： 有一台服务器，服务器上有多块儿GPU能够供使用，但此时只但愿使用第2块和第4块GPU，可是咱们但愿代码能看到的仍然是有两块GPU，分别编号为0,1，这个时候咱们能够使用环境变量CUDA_VISIBLE_DEVICES来解决这个问题。 好比：linux CUDA_VISIBLE_DEVICES=1 只有编号为1的GPU对程序是可见的，在代码中gpu[0]指的就是这块儿GPU CUDA_VIS

>>阅读原文<<