本文共 1110 字,大约阅读时间需要 3 分钟。
编译并使用PyTorch的项目需要注意Visual Studio版本与CUDA版本的匹配,同时确保依赖库正确加载。以下是从你提供的信息整理出来的优化后的步骤,希望对你有所帮助:
检查Visual Studio版本:确保安装的是Visual Studio 2017 15.6或更高版本,并确认《MSC VER》≤1912。可以通过右键点击项目文件,选择“属性”,查看工具set版本。
升级CUDA工具set:如果需要支持CUDA 10.x版本,确保安装了CUDA 10.2及其以上版本的工具set。可以通过NVIDIA的网站下载相应的安装包。
下载适配的libtorch:在PyTorch官网下载对应CUDA版本和Visual Studio版本的预编译libtorch库。例如,若使用CUDA 10.2和Visual Studio 2017 15.6,可以下载对应的libtorch1.5.1-cu10.2.
配置编译环境:
CMAKE_GENERATOR_TOOLSET_VERSION
为14.11,以匹配你的Visual Studio工具set。编译libtorch:
cmake --build .
,生成编译结果。生成测试项目代码:
torch/script.h
和at.h
)。torch::jit::load
加载.pt
文件。处理依赖库冲突:
c10.dll
、torch_cpu.dll
)已经被正确链接。调试和优化:
验证测试/demo的运行性:
通过逐步检查每个环节,包括依赖设置、编译选项、动态库路径等,应该可以解决主流问题。遇到问题时,参考社区讨论或PyTorch官方文档,确保配置与环境完全兼容。
转载地址:http://dtqiz.baihongyu.com/