目录
环境:
windows 10;
vs2019;
libtorch1.8.1-cuda10.2
opencv3.4.14
cmake 3.20.2
前期准备:
libtorch下载(Release版):
在https://pytorch.org/get-started/locally/中寻找符合自己的版本,我的选择如下:
建议使用release version,debug version未实测过。
这里的pytorch版本需和训练yolov5的pytorch版本统一,如果需要其他版本的libtorch,推荐看这篇博客。
我使用的版本的地址如下:
gpu:https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.8.1%2Bcu102.zip
cpu:https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.8.1%2Bcpu.zip
一般只下载gpu版即可,除非你只需在cpu上跑模型。
下载完成后解压在任意地址即可,这里假定文件在D:\libtorch中。libtorch文件夹内容如下。
opencv下载:
在https://opencv.org/releases/中选择自己所需版本即可。我这里选择的是3.4.14的windows版。https://sourceforge.net/projects/opencvlibrary/files/3.4.14/opencv-3.4.14-vc14_vc15.exe/download
双击执行,解压在任意地址即可,这里假定文件在D:\opencv中,文件夹内容如下。
cuda10.2
下载cuda10.2并安装,这类教程就很多了。安装完成后在cmd中输入nvcc -V,若能成功输入版本号即为成功。
下载cudnn(最新版本即可),并按教程放在cuda文件夹内。
环境变量设置
安装完成后,右键“此电脑”->“属性”->“高级系统设置”->“环境变量”。对用户变量中的Path进行修改
加入"D:\libtorch\lib",“D:\opencv\build\x64\vc15\