nvidia-smi
查看当前节点下有几块GPU,以及每一块的使用情况
CUDA_VISIBLE_DEVICES=1 python main.py --gpu-devices 1
在运行程序时指定使用哪块GPU(先cd进入main.py所在文件夹)
nvidia-smi
查看当前节点下有几块GPU,以及每一块的使用情况
CUDA_VISIBLE_DEVICES=1 python main.py --gpu-devices 1
在运行程序时指定使用哪块GPU(先cd进入main.py所在文件夹)