文章目录
关于PyTorch
PyTorch是一个Python包,提供了两个高级特性:
-强大的GPU加速张量计算(如NumPy)
-深度神经网络建立在一个基于磁带的autograd系统
您可以在需要时重用您喜欢的Python包(如NumPy、SciPy和Cython)来扩展PyTorch。
关于LibTorch
从PyTorch版本迭代到1.0以后,Facebook就更新了Pytorch支持C++导入模型的torch.jit
API, 用于生成支持C++ API的模型文件。
安装
- 下载安装包
# Mac (不支持CUDA)
wget https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip
unzip libtorch-macos-1.4.0.zip
# Linux (CUDA10.1)
wget https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.4.0.zip
unzip libtorch-shared-with-deps-1.4.0.zip
# Linux (CPU)
wget https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.4.0+cpu.zip
unzip libtorch-shared-with-deps-1.4.0+cpu.zip
2.执行脚本文件
保存为install.sh
至解压目录下
# copy torch header file
sudo cp -r include/* /usr/local/include
# copy torch library file
sudo cp -r lib/* /usr/local/lib
# copy torch share file
sudo cp -r share/* /usr/local/share
卸载
1.执行脚本文件
保存为uninstall.sh
至解压目录下</