cuda指定GPU,在terminal中指定

nvidia-smi

查看当前节点下有几块GPU,以及每一块的使用情况

CUDA_VISIBLE_DEVICES=1 python main.py --gpu-devices 1

在运行程序时指定使用哪块GPU(先cd进入main.py所在文件夹)

你可能感兴趣的:(令人头大的环境问题,神经网络,人工智能,机器学习)