https://github.com/christian-rauch/torch_cpp
CMake wrapper for installing LibTorch (PyTorch C++ API)
https://github.com/christian-rauch/torch_cpp
cmake pytorch
Last synced: 3 months ago
JSON representation
CMake wrapper for installing LibTorch (PyTorch C++ API)
- Host: GitHub
- URL: https://github.com/christian-rauch/torch_cpp
- Owner: christian-rauch
- Created: 2020-07-06T12:34:02.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2025-06-17T09:05:20.000Z (6 months ago)
- Last Synced: 2025-06-17T09:47:43.957Z (6 months ago)
- Topics: cmake, pytorch
- Language: CMake
- Homepage: https://pytorch.org/cppdocs/
- Size: 8.79 KB
- Stars: 10
- Watchers: 1
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# LibTorch install script for CMake
This CMake project is basically an install script for LibTorch (PyTorch C++ API). It downloads the binary archieve and installs headers, libraries and CMake configurations files to the corresponding `INCLUDEDIR`, `LIBDIR` and `DATADIR` directories.
The package can be used as part of a pure CMake catkin or colcon workspace by including:
```XML
cmake
torch_cpp
cmake
```
in your `package.xml`. The relevant files will then be installed within the workspace target folder (`install` or `devel`).
Usage in a CMake project:
```CMake
find_package(Torch REQUIRED)
target_link_libraries(${PROJECT_NAME} PUBLIC torch)
set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 14)
target_link_options(${PROJECT_NAME} PUBLIC ${TORCH_CXX_FLAGS})
```
If you get the error `No CMAKE_CUDA_COMPILER could be found.`, then the CUDA compiler `nvcc` cannot be found in the default search paths (`$PATH`). In this case, you have to set the path to `nvcc` manually:
```sh
export CUDACXX=/usr/local/cuda/bin/nvcc
```