设置环境变量指定使用的 GPU

先

nvidia-smi

看看使用情况

如：

再

export CUDA_VISIBLE_DEVICES=0,1,2

指定使用的 GPU

posted @ 2024-03-03 20:49  Cold_Chair  阅读(22)  评论(0)  编辑  收藏  举报