pytorch模型训练指定显卡

pytorch版本+cuda版本匹配

测试cuda是否可用

保存模型的几种格式.pt.pth.pkl：它们并不是在格式上有区别，只是后缀不同而已（仅此而已）保存模型的两种方式方式一：只保存模型权重参数，不保存模型结构torch.save(model.state_dict(), 'mymodel.pth') # 只保存模型权重参数，不保存模型结构model = My_model(*args, **kwargs) # 这里需要重新模型结构，My_modelmo

数据集标注