Pytorch遇到的几个问题

pytorch 专栏收录该内容
0 篇文章 0 订阅

本文主要记录一下使用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 *

保存并退出,重新运行即可

  • 3
    点赞
  • 0
    评论
  • 3
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值