pytorch学习知识点总结

查看torch版本、是否安装cuda、查看cuda版本

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)

1 torch.save(): 保存一个序列化（serialized）的目标到磁盘。
2 torch.load(): 用来加载模型。
3 torch.nn.Module.load_state_dict(state_dict) 用来加载模型参数。