查看pytorch版本(通过PyCharm查看):
Cuda版本查看:
2:下载Libtorch1.10.0。
https://blog.csdn.net/weixin_43742643/article/details/115218126
注:Libtorch的版本必须和你训练使用的虚拟环境中的pytorch版本一致。下载之前先去你的虚拟环境中查看Pytorch版本。
(1) 选择对应的版本
(2) 选择Release or Debug (我用的Debug)
(3) 选择CPU or CUDA(GPU)
3:下载Opencv
Opencv下载地址:https://opencv.org/releases/
4:下载Libtorch-Yolov5
下载地址:https://github.com/yasenh/libtorch-yolov5
5:下载Cmake
下载地址:https://cmake.org/download/
6.解决libTorch中Torch::cuda::is_available()返回为False
https://blog.csdn.net/qq_36038453/article/details/120278523