YOLOV5环境torchvision和pytorch的版本不一致

hzhe

已经配置好环境有半年了，半年没怎么用，现在想用，发现出问题了。

torchvision和pytorch的版本不一致

1、看看torchvision支持的CUDA版本

首先，打开Anaconda Prompt(因为之前是通过anaconda配置的名为yolov5虚拟环境），开始菜单，打开就可以看到了

输入

conda activate yolov5

 打开名为yolov5虚拟环境

输入python回车

​
import torch

print(torch.version.cuda)

​

发现确实版本不匹配。（我在这里走入了误区，因为本人电脑配置的是CUDA11.6版本，由于另一个项目需要这个版本。于是考虑装两个版本的CUDA来配置不同的项目。但是太麻烦了。）又经过搜索，

1、进入Anaconda Prompt，通过conda activate yolov5进入yolov5虚拟环境卸载pytorch ,torchvision

pip uninstall torch
pip uninstall torchvision

2、进入pytorch官网

选择对应的需要的版本，复制命令行下载。

conda install pytorch torchvision torchaudio cudatoolkit=11.3

不要复制 -c pytorch 

问题解决。（虚拟环境真好用）

找不到就点

然后找对应的版本

问题解决