AssertionError: Invalid CUDA ‘--device 0,1,2‘ requested, use ‘--device cpu‘ or pass valid CUDA devic

错误：AssertionError: Invalid CUDA '--device 0,1,2' requested, use '--device cpu' or pass valid CUDA device(s)
在运行yolov5时，出现这个错误，意思是没有可用的cuda，无法使用GPU训练。
1.首先用nvidia-smi查看是否真的有三张显卡。很有可能是没有这么多的。这里面的cuda–version的版本是最高支持的cuda的版本，比这个低的是可以用的。
2.检查自己安装的pytorch是不是装错了，可能装到了cpu的版本。这时候就算print(torch.cuda.is_available())是True也是不可用的。