GPU训练Cuda报错

        原本进行了YOLOV5的图片分类实验任务，成功配置了pytorch的CUDA 10.2的GPU训练环境，今天在进行YOLOV5目标检测任务时似乎因为某个报错升级了torchvision的版本到0.16，导致训练过程中报错:

AssertionError: Invalid CUDA '--device  0' requested, use '--device cpu' or pass valid CUDA device(s)

解决办法：

（1）先是查看自己电脑安装的cuda版本：

nvcc -V

（2） 去pytorch previous-versions网站上查找对应自己cuda版本的pytorch

卸载torchvision

conda uninstall torchvision

重新下载安装对应版本torchvision

conda install torch==1.9.0+cu102 torchvision==0.10.0+cu102

注： Wheel版本的下载会非常快！推荐！！！

4060Ti:

pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121

（3）检测是否安装成功

$: source activate torch（环境名）
或：conda activate torch
$: python
import torch
torch.cuda.is_available()