CUDA程序在运行的时候如何选择可用的计算设备

来源:互联网 发布:国玺云计算有限公司 编辑:程序博客网 时间:2024/04/29 08:40

CUDA_VISIBLE_DEVICES 这个环境变量在Linux下可以很好的完成这个任务。

如果你使用devicequery发现的设备顺序是:

0:GTX580

1:K20C

2:K20C

如果你想让CUDA程序只可以使用1号和2号显卡,可以这样设置:

CUDA_VISIBLE_DEVICES=1,2

或者

CUDA_VISIBLE_DEVICES =“1,2”


快去试试。。

原创粉丝点击