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

hzhe

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

torchvision和pytorch的版本不一致

1、看看torchvision支持的CUDA版本

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

YOLOV5环境torchvision和pytorch的版本不一致_第1张图片

输入

conda activate yolov5

 打开名为yolov5虚拟环境

输入python回车

​
import torch

print(torch.version.cuda)

​

YOLOV5环境torchvision和pytorch的版本不一致_第2张图片

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

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

pip uninstall torch
pip uninstall torchvision

2、进入pytorch官网

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

YOLOV5环境torchvision和pytorch的版本不一致_第3张图片

conda install pytorch torchvision torchaudio cudatoolkit=11.3

不要复制 -c pytorch 

YOLOV5环境torchvision和pytorch的版本不一致_第4张图片

YOLOV5环境torchvision和pytorch的版本不一致_第5张图片

问题解决。(虚拟环境真好用)

找不到就点

YOLOV5环境torchvision和pytorch的版本不一致_第6张图片

然后找对应的版本

YOLOV5环境torchvision和pytorch的版本不一致_第7张图片YOLOV5环境torchvision和pytorch的版本不一致_第8张图片

YOLOV5环境torchvision和pytorch的版本不一致_第9张图片

问题解决

 

你可能感兴趣的:(目标检测)