【WSL】【libtorch】跑起来

TTMer

于 2022-03-27 12:31:32 发布

阅读量1k

点赞数

分类专栏： WSl libtorch C++ 文章标签：深度学习 c++

本文链接：https://blog.youkuaiyun.com/qq_39632866/article/details/123770922

版权

😍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