首先根据自己的CUDA
版本安装正确对应的cuda
和cudnn
并进行配置。
这里安装的是cuda-11.3
版本,以下基于这个版本进行安装。
1. 安装 Clion
因为Clion
更容易直接编写CMakelists.txt
,所以使用Clion
作为IDE。
需要在File -> Setting -> CMake
的CMake options
中添加
-DCMAKE_CUDA_ARCHITECTURES=86 -DCMAKE_CUDA_COMPILER=”/usr/local/cuda/bin/nvcc”
不然在编译时会出现“CMAKE_CUDA_ARCHITECTURIES must be non-empty if set.”
的错误。
2. 下载 Libtorch
注意libtorch
的版本一定要和pytorch
的版本完全相同。可根据Libtorch各版本下载进行下载,这里下载的是对应的1.11.0
版本
https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.11.0%2Bcu113.zip
3. 下载 OpenCV
注意opencv
的版本不能高于4.0.0
,不然在编译的时候会因为-D_GLIBCXX_USE_CX11_ABI=0
问题与libtorch
产生冲突。所以不要使用sudo apt-get install -y libopencv-dev
直接安装,会下载最新版本。这里安装3.4.20
版本
首先进行依赖包的下载
sudo