Pytorch遇到的几个问题

xiaoxiao2021-02-28  180

本文主要记录一下使用pytorch几个问题

问题一

ImportError: No module named torchvision

torchvison:图片、视频数据和深度学习模型

解决方案

安装torchvision,参照官网

问题二

安装torchvision过程中遇到 Could not find a version that satisfies the requirement olefile (from pillow->torchvision==0.1.8) 

没有olefile

解决方案

安装olfile,官网

问题三

找不到nn.init模块

具体错误内容 'module' object has no attribute 'init'

解决方案

找到以下目录torch/nn, 此目录下有init.py和__init__.py文件,打开__init__.py文件看到以下内容:

from .modules import * from .parameter import Parameter from .parallel import DataParallel在代码的最后加上以下语句

from .init import * 保存并退出,重新运行即可

转载请注明原文地址: https://www.6miu.com/read-33497.html

最新回复(0)