在使用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
