模型训练报错AttributeError: module 'torch.storage' has no attribute 'TypedStorage'怎么办
时间: 2024-06-03 13:06:05 浏览: 156
开发errorjsp-手把手带你入门ssh项目
这个错误通常是由于 PyTorch 版本问题导致的,具体来说是因为 PyTorch 1.7.0 及以上版本不再支持 TypedStorage。解决这个问题的方法是升级 PyTorch 到最新版本或者降级到 PyTorch 1.6.0 版本以下。
如果你想升级 PyTorch 版本,可以使用以下命令:
```
pip install torch -U
```
如果你想降级 PyTorch 版本,可以使用以下命令:
```
pip install torch==1.6.0
```
另外,还有一些其他的解决方法,比如重新安装 PyTorch、检查代码中是否有使用到 TypedStorage 等,但是在大多数情况下升级或者降级 PyTorch 版本就能够解决这个问题。
阅读全文