CUDA_VISIBLE_DEVICES 这个环境变量在Linux下可以很好的完成这个任务。
如果你使用devicequery发现的设备顺序是:
0:GTX580
1:K20C
2:K20C
如果你想让CUDA程序只可以使用1号和2号显卡,可以这样设置:
CUDA_VISIBLE_DEVICES=1,2
或者
CUDA_VISIBLE_DEVICES =“1,2”
快去试试。。
CUDA_VISIBLE_DEVICES 这个环境变量在Linux下可以很好的完成这个任务。
如果你使用devicequery发现的设备顺序是:
0:GTX580
1:K20C
2:K20C
如果你想让CUDA程序只可以使用1号和2号显卡,可以这样设置:
CUDA_VISIBLE_DEVICES=1,2
或者
CUDA_VISIBLE_DEVICES =“1,2”
快去试试。。