怎么查看pth文件的pytorch版本
时间: 2024-05-04 13:20:14 浏览: 201
EasyOCR训练好的模型(包含english-g2.pth、zh-sim-g2.pth、craft-mlt-25k.pth)
5星 · 资源好评率100%
你可以使用以下代码来查看pth文件所对应的PyTorch版本:
```python
import torch
model = torch.load('your_model.pth', map_location=torch.device('cpu'))
print('PyTorch version:', torch.__version__)
print('Model PyTorch version:', model['pytorch_version'])
```
在这个示例中,我们使用`torch.load()`函数加载了保存在`your_model.pth`中的模型,并使用`map_location=torch.device('cpu')`参数指定了CPU作为设备。然后,我们打印了当前安装的PyTorch版本和模型中保存的PyTorch版本。
阅读全文