Ubuntu20.04+vscode的libtorch环境配置(跑起来测试demo)
Ubuntu20.04+vscode的libtorch环境配置(跑起来测试demo)初学,网上资料五花八门,又多又乱,查了很久资料才跑通,记录一下防止自己老年痴呆…1.在pytorch官网找到libtorch,下载压缩文件,解压获得源码官网下载2.使用pytorch训练模型(前面可以都不看,就看模型怎么保存的)if __name__ == '__main__': # 训练好的模型 model = torch.load('model/best_model.pth') # 必须使用t
原创
2021-09-04 20:32:53 ·
1662 阅读 ·
0 评论