问题: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下载地址