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

先

nvidia-smi

看看使用情况

如：

再

export CUDA_VISIBLE_DEVICES=0,1,2

指定使用的 GPU

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