解决：AssertionError: CUDA unavailable, invalid device gpu requested

因为是刚接触机器学习，搞cpu的已经搞定，换gpu却出问题了，环境搞了一天。各种百度查。看到这里的朋友不要百度查了，直接上官网，上不了官网的参考一下方式：按照官网给的依赖包版本安装，不然容易踩依赖包版本不合适导致问题

PyTorch官网

注：官网不断更新，安装失败建议去官网看 下载地址更新于：2020-03-09
CUDA 10.2 官网下载
CUDA 11.3 官网下载

PyTorch官网 给出的 CUDA 10.2 安装依赖

Conda安装：conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
pip安装：pip3 install torch1.10.2+cu102 torchvision0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

PyTorch官网 给出的 CUDA 11.3 安装依赖

Conda安装：conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
CUDA 11.3
pip 安装：pip3 install torch1.10.2+cu113 torchvision0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

最近又遇到了一个新的问题：一位好友私信我，最后我看他的环境，环境没有任何问题，重建了一个文件print(torch.cuda.is_available)输入为True。但是在train.py中输入为False。最后吧train.py文件名改掉可以了，问题就这样解决了。神奇不神奇，遇到这个问题的朋友可以试试先

对比图片：

注：我遇到了这样一个问题，最后可以运行了不知为啥。环境是yolov5， anaconda虚拟环境 python 3.8.8

单独运行torch是可以的

运行yolov5的 detect.py 却报错

最可笑的是在运行整个识别之前导入torch之后马上调用一下：torch.cuda.is_available() 竟然可以了：表示不懂为什么有大佬懂得给指点一下