通过设置CUDA_VISIBLE_DEVICES环境变量来指定使用哪些GPU设备
发布时间:2024年01月09日
指定单个GPU设备:
CUDA_VISIBLE_DEVICES=0
指定多个GPU设备:
CUDA_VISIBLE_DEVICES=0,1,2
指定一个范围的GPU设备:
CUDA_VISIBLE_DEVICES=0-2
在设置了CUDA_VISIBLE_DEVICES环境变量后,只有被指定的GPU设备会被程序所使用,其他未被指定的GPU设备将不可见。
注意:这种方式只对使用CUDA进行加速的程序有效,对于仅使用CPU的程序无效。
文章来源:https://blog.csdn.net/qq_16792139/article/details/135477898
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:chenni525@qq.com进行投诉反馈,一经查实,立即删除!