安装与使用no cuda版本libtorch
时间: 2024-05-02 07:16:35 浏览: 253
安装no cuda版本的libtorch可以通过以下步骤完成:
1. 打开libtorch官网,选择对应的版本和操作系统,下载no cuda版本的libtorch压缩包。
2. 解压缩下载的压缩包到指定目录,例如解压缩到 /usr/local/libtorch 目录下。
3. 设置环境变量,将 libtorch 的 bin 目录和 lib 目录加入到 PATH 和 LD_LIBRARY_PATH 中,例如:
```
export PATH=/usr/local/libtorch/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/libtorch/lib:$LD_LIBRARY_PATH
```
4. 在 C++ 项目中包含 libtorch 的头文件和链接 libtorch 库即可使用。
使用no cuda版本的libtorch时,需要注意以下几点:
- 不支持 GPU 加速,只能在 CPU 上运行。
- 需要使用 CPU 版本的 OpenBLAS 库,而 GPU 版本的 OpenBLAS 库只能用于 cuda 版本的 libtorch。
- 编译 C++ 项目时需要使用 -DC10_USE_CPU_ONLY=ON 选项开启 CPU 模式,或者在 CMakeLists.txt 文件中添加如下代码:
```
set(CMAKE_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=0 ${CMAKE_CXX_FLAGS}")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O2 -DNDEBUG -fPIC -fvisibility-inlines-hidden -Wno-deprecated -Wno-deprecated-declarations -Wno-error=sign-compare -Wno-error=unused-result -Wno-strict-aliasing")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-missing-braces -Wno-parentheses -Wno-unused-function")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-non-virtual-dtor -Wno-overloaded-virtual -Wno-exit-time-destructors")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14 -fopenmp")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DC10_USE_CPU_ONLY=ON")
```
阅读全文