通过设置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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。