1. 根据显卡版本下载libtorch
例如cuda11.3+torch1.12.0: 点击下载
2. visual studio 2022 配置属性表
-
属性管理器->Release->右键添加新项目属性表
-
双击打开属性表,添加头文件目录
2.1 c/c++ -> 常规 -> 附加包含目录, 添加:
xxx\libtorch\include
xxx\libtorch\include\torch\csrc\api\include
(xxx
是你解压所在的路径)
2.2 添加库目录。链接器 -> 常规 -> 附加库目录,添加:
xxx\libtorch\lib
2.3 添加库文件。链接器 -> 输入 -> 附加依赖项。将lib
文件夹下所有的lib
文件都添加进去。可以在当前目录下的cmd
窗口中使用命令dir /b *.lib>1.txt
生成txt文件,将txt文件的内容全部复制粘贴到附加依赖项里面。
2.4 添加命令行。链接器 -> 命令行 -> 其他选项, 复制以下内容粘贴进去:/INCLUDE:?warp_size@cuda@at@@YAHXZ /INCLUDE:?_torch_cu