Libtorch libtorch详解

1. 必要软件

  • wget
  • unzip
  • python3
  • pip3
  • cmake
  • gcc (or clang)
  • make (or ninja)

软件版本尽量用新的。

2. 安装 PyTorch

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu

推荐使用 miniconda 管理 python 环境。

3. 下载解压 LibTorch

wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
unzip libtorch-shared-with-deps-latest.zip

注意1:本教程适用 Linux 系统,Windows 等我有时间,MacOS 等我有钱。

注意2:以上均是 CPU 版本,有 GPU 并配置了 CUDA 的土豪移步官网根据版本下载。

4. 配置环境

三种方式(影响范围由大到小):

  1. 将讲解压目录 /path/to/libtorch(注意替换)添加到系统 PATH 变量中:
# 临时使用
export PATH=/path/to/libtorch:$PATH

# 永久使用
echo "export PATH=/path/to/libtorch:\$PATH" >> ~/.bashrc && . ~/.bashrc
  1. 设置环境变量 Torch_ROOT,方法参考上面。
  2. CMakeLists.txt 中通过 set 函数临时设置。
  3. 执行 cmake 命令的时候,设置参数 -DCMAKE_PREFIX_PATH=/path/to/libtorch

推荐使用第2种。

5. 运行 demo

下载项目

git clone https://github.com/clearhanhui/LearnLibTorch.git
cd LearnLibTorch/chap1/

使用 make 构建

mkdir build-make && cd build-make
cmake .. 
make
./HelloWorld

使用 ninja 构建

mkdir build-ninja && cd build-ninja
cmake .. -G Ninja
ninja -v
./HelloWorld

输出:

 0  0  0
 0  0  0
[ CPUFloatType{2,3} ]

Welcome to LibTorch!

教程来源于Github,感谢clearhanhui大佬的无私奉献,致敬!

技术教程推荐

人工智能基础课 -〔王天一〕

从0开始学架构 -〔李运华〕

从0打造音视频直播系统 -〔李超〕

浏览器工作原理与实践 -〔李兵〕

雷蓓蓓的项目管理实战课 -〔雷蓓蓓〕

Kafka核心源码解读 -〔胡夕〕

爱上跑步 -〔钱亮〕

朱涛 · Kotlin编程第一课 -〔朱涛〕

云原生架构与GitOps实战 -〔王炜〕