Pytorch1.7.1和LibTorch的安装

ubuntu: 18.04
Nvidia DriverVersion: 430.64
CUDA Version: 10.1
Cudnn: 7.6.5
Pytorch: 1.7.1
torch: 1.7.1

驱动安装

先进行驱动的安装

驱动安装见https://chenbokai.icu/tech/carla_1_step/

目前只能安装430版本

cuda安装

cuda对应版本通过nvidia-smi查看

安装时需注意将驱动安装去掉,否则会出现驱动版本不适配无法启动的问题

配置环境变量

sudo vi ~/.bashrc
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
sudo source  ~/.bashrc

然后利用nvcc -V可以查看输出,确认安装成功

cudnn安装

sudo cp cuda/include/cudnn.h	/usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn*	/usr/local/cuda/lib64/

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

安装版本为7.6.5 ,8版本以后安装后不能成功打印输出 version.h的文件未找到

Pytorch安装

安装libtorch之前,不按照pytorch 可能无法使用无法找到caffe2的编译库问题

安装对应cuda的torch版本 pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

安装过程中有一个问题因为zlib未安装导致Pillow无法解压 Pillow无法解压导致torchvision安装失败.

python3 
import torch
import torchvision
print(torch.cuda.is_available())//验证GPU

libtorch安装

libtorch下载地址 https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.7.1%2Bcu101.zip

使用官方例子进行验证

CMakeLists.txt

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(example-app)
set(CMAKE_PREFIX_PATH "绝对路径/libtorch")
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")

add_executable(example-app example-app.cpp)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 14)

example-app.cpp

#include <torch/torch.h>
#include <iostream>

int main() {
  torch::Tensor tensor = torch::rand({2, 3});
  std::cout << tensor << std::endl;
}

运行后效果

下载.jpeg

Libtorch在linux下的安装