遇到的问题:
用网上的教程用 pip 安装torch的时候可以运行不报错 并且 可以在 pycharm 中能 import torch 。
但是安装 torchvision 在anaconda虚拟环境下没有显示,所以不能再pycharm中import 。
ModuleNotFoundError: No module named 'torchvision'
问题之前的工作:
我按照安装教程下载了 torch 和 torchvision的whl 文件。
并成功pip了 torch,在pycharm中import 不报错。
按照本地路径 pip torchvision 成功,但是anaconda环境中没有显示。
解决:
主要是解决torchvision的问题。
在这里,首先 查找官网的介绍:https://pypi.org/project/torchvision/0.1.8/
也就是说可以有三种安装方式,这里开始我先用了 pip 的方法,导入成功但是在anaconda 里并不可以直接使用,失败。所以,我采用第一种方式。
在管理员模式下,直接输入了这行代码:conda install torchvision -c soum