ModuleNotFoundError: No module named ‘torchvision.ops‘

问题:cuda的版本是cuda 9.0,在使用torchvision=0.2.1,torch版本为0.4.0时候报错:

在这里插入图片描述

原因:torchvision和cuda版本不匹配,torchvision的版本过低,应该将其至少更新到0.3.0。
解决办法

将torchvision0.2.1更新到torchvision0.4.0

卸载

 pip uninstall torchvision

安装
推荐离线安装torchvision,到下面这个地址找到自己的安装版本(我的是windows系统和python3.6),把torchvision对应版本的whl文件直接下载到本地。pip install xxx.whl 安装包即可。
torchvision下载地址
ModuleNotFoundError: No module named ‘torchvision.ops‘_第1张图片

ModuleNotFoundError: No module named ‘torchvision.ops‘_第2张图片

你可能感兴趣的:(深度学习,深度学习,pytorch,神经网络)