查看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) 用來加載模型參數。

總結

以上是生活随笔為你收集整理的pytorch学习知识点总结的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。