【WSL】【libtorch】跑起来

CMake参数
示例程序
cuda 安装
cudnn 安装
 libtorch 安装

CMake 参数

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(main)
set(Torch_DIR /home/whtower/CppLibs/libtorch/share/cmake/Torch)
set(CMAKE_CUDA_COMPILER /usr/local/cuda-11.3/bin/nvcc)
find_package(Torch REQUIRED)
add_executable(main main.cpp)
target_link_libraries(main "${TORCH_LIBRARIES}")
set_property(TARGET main PROPERTY CXX_STANDARD 14)

示例程序

#include 
#include 
int main(int argc, char **argv)
{
    torch::Tensor t = torch::eye(3);
    std::cout << t << std::endl;
    return 0;
}

cuda 安装

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/cuda.list'
sudo apt-get update
sudo apt-get install -y cuda-toolkit-11-3

对应系统版本和cudatoolkit版本在相应位置修改(ubuntu2004)(cuda-toolkit-11-3)
查看系统版本

lsb_release -a

cudnn 安装
nvidia官网下载对应cudnn
本地打开进行解压

tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz # 进行解压
sudo cp cuda/include/cudnn*.h    /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn*    /usr/local/cuda-11.3/lib64
sudo chmod a+r /usr/local/cuda-11.3/include/cudnn*.h   /usr/local/cuda-11.3/lib64/libcudnn*

# 查看版本
cat /usr/local/cuda-11.3/include/cudnn.h | grep CUDNN_MAJOR -A 2
# cudnn 8 以上 查看版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

libtorch 安装
pytorch 官网下载

上面那个是早期c++版本
下面那个是新c++版本，一般都是下面那个
下载完解压放着就行了，在CMake配置时链接相应的目录