mmdetection训练 显卡选择

训练细节

在终端执行.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开始）