libtorch是pytorch的C++库,常用于pytorch模型的C++部署。
1、cuda安装
参考:
windows/linux安装NVIDIA驱动(cuda加速)-CSDN博客https://blog.csdn.net/m0_69115733/article/details/159944127?spm=1001.2014.3001.5502
2、libtorch下载
libtoch是在pytorch官网下载
https://pytorch.org/https://pytorch.org/
安装cuda版的libtorch时,注意cuda版本和自己电脑匹配!!!