Pytorch 1.0.0版本问题之module ‘torch.nn‘ has no attribute ‘Flatten‘

  1. 原因:pytorch1.0.0没有Flatten类
  2. 解决方法(有效!!,请用pytorch1.9.0下的文件进行替换)

pytorch1.9.0链接:

链接:https://pan.baidu.com/s/111YZm3gFEGlkt3FoYeVj4A 
提取码:ecnu

  1. 第一步,找到本地安装pytorch的路径,我是通过ANACONDA安装,路径如下

E:\2345Downloads\Anaconda\anaconda\envs\pytorch

找到以下路径

E:\2345Downloads\Anaconda\anaconda\envs\pytorch\Lib\site-packages\torch\nn\modules

对其中的__init__文件修改

from .flatten import Flatten, Unflatten

 

2.第二步,替换文件

解压找到目录D:xx\pytorch-1.9.0\pytorch-1.9.0\torch\nn\modules中的flatten.py文件复制到自己电脑安装的相同路径下

3.第三步,替换第二个文件

找到你下载的1.9.0的文件下的D:xx\pytorch-1.9.0\torch的types.py文件

也放到原来的同样的位置

E:\2345Downloads\Anaconda\anaconda\envs\pytorch\Lib\site-packages\torch

4.最后,修改刚刚复制过来的types.py文件的第21行,注释掉这行 

_qscheme = torch.qscheme将加上其‘#’屏蔽。

你可能感兴趣的:(pytorch,深度学习,人工智能)