AssertionError: CUDA unavailable, invalid device 0 requested

在使用yolov5训练时，提示这个错误，查看torch.cuda.is_available()得到的返回值是false

查看nvidia-smi可以看到信息

由于之前使用conda安装的torch1.9.0

这里使用pip重新安装：

pip3 install torch == 1.9.0+cu111 torchvision == 0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

全部安装后，再次执行torch.cuda.is_available()时

可以看到正确结果了

参考：AssertionError: CUDA unavailable, invalid device 0 requested_消灭BUG鸭的博客-CSDN博客_cuda unavailable