在利用torchvision包处理图象数据集时,我们发现产生了错误:ModuleNotFoundError: No module named 'torchvision',这说明我们可能未安装torchvision包。
目录
一、安装torchvision
1.1 检查是否安装:
打开Anaconda的Anaconda Prompt,输入:
conda list
查看是否安装torchvision。
1.2 未安装时安装torchvision
我们这里并没有安装torchvision,即可直接安装,输入:
pip install torchvision==0.4.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
后面 -i 之后的是选择豆瓣源,防止安不上,安装的是0.4.1的torchvision。当然不选择版本直接安装也可以,即:
pip install torchvision -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
安装结果为:
1.3 已安装torchvision版本过低
只需要先卸载torchvision,输入:
pip uninstall torchvision
即可卸载torchvision,后面按1.2的内容安装一遍就行。
二、安装torchaudio、torchtext
另外,我们常用的数据集载入还有torchaudio、torchtext,可以仿照此例进行安装:
安装torchaudio代码为:
pip install torchaudio -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
安装torchtext代码为:
pip install torchtext -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
安装torchaudio结果为:
安装torchtext结果为:
来源地址:https://blog.csdn.net/weixin_43907136/article/details/127030361