在终端执行.py文件时通过CUDA_VISIBLE_DEVICES来指定:
在终端运行python程序的前面指定,比如:
CUDA_VISIBLE_DEVICES=1 python train.py
实验效果如下:
在终端运行脚本或者jupyter notebook下运行:
前面加上CUDA_VISIBLE_DEVICES=1
watch -n 1 -d nvidia-smi
就会实时刷新了,可以查看显卡的用量等,哪块卡是空闲的(0开始)