问题描述
使用torchvision库时,报错无法正常运行
问题原因和解决方式
torch和torchvision版本不匹配
1.按照官网安装对应cuda版本的pytorch
- 进入pytorch官网
https://pytorch.org/ - 选择你的cuda版本
- 复制红框里的指令去你的虚拟环境里安装
*也就是说可能你需要提前执行 conda activate your-env-name
2.手动安装匹配版本
- 打开tochvision文档
https://pypi.org/project/torchvision/
- 确定自己需要安装的版本
- 根据自己对应版本进行安装