MANFOLD TX2配置Pytorch1.x方法總結（Python3.5+CUDA9.0+Opencv3.4.1）

硬件環境：

MANFOLD TX2

Ubuntu16.04

Python 3.5

CUDA 9.0

CDUANN 7.x

配置過程：

1、查看CUDA與CUDANN版本

dji@manifold2:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Sun_Nov_19_03:16:56_CST_2017
Cuda compilation tools, release 9.0, V9.0.252

dji@manifold2:~$ ldconfig -p | grep dnn
        libopencv_dnn.so.3.3 (libc6,AArch64) => /usr/lib/libopencv_dnn.so.3.3
        libopencv_dnn.so (libc6,AArch64) => /usr/lib/libopencv_dnn.so
        libcudnn.so.7 (libc6,AArch64) => /usr/local/cuda-9.0/lib64/libcudnn.so.7
        libcudnn.so.7 (libc6,AArch64) => /usr/lib/aarch64-linux-gnu/libcudnn.so.7
        libcudnn.so (libc6,AArch64) => /usr/lib/aarch64-linux-gnu/libcudnn.so

2、更換TX2國內鏡像

 【注意】下載的是ubuntu-ports後綴的鏡像源，目前所知中科大的源可用，其他嘗試均失敗），sudo apt-get install找不到對應的包，無法更新系統軟件包和環境依賴。

deb http://mirrors.ustc.edu.cn/ubuntu-ports/  xenial main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports/  xenial-updates main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports/  xenial-security main restricted universe multiverse
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/  xenial main restricted universe multiverse
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/  xenial-updates main restricted universe multiverse
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/  xenial-security main restricted universe multiverse

3、安裝環境依賴

sudo apt install libopenblas-dev libatlas-dev liblapack-dev
sudo apt install liblapacke-dev checkinstall
sudo apt-get install python3-pip
pip3 install --upgrade pip
sudo apt-get install python3-dev
sudo pip3 install numpy scipy
sudo pip3 install pyyaml
sudo pip3 install scikit-build
sudo apt-get -y install cmake
sudo apt install libffi-dev
sudo pip3 install cffi

4、添加CUDA和CUDANN路徑

sudo gedit ~/.bashrc
#add CUDA path to bashrc
export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu
export CUDNN_INCLUDE_DIR=/usr/include
source ~/.bashrc

5、pip3 安裝wheel

感謝torch-1.0.0a0+db5d313-cp35-cp35m-linux_aarch64.whl該github主的無私奉獻【開源萬歲 *\ ^ ~ ^ /* 】

下載該whl文件，到指定目錄下，執行以下命令：

sudo pip3 install torch-1.0.0a0+db5d313-cp35-cp35m-linux_aarch64.whl

兩分鐘過後你將看到提示:successfully install pytorch

6、測試pytorch是否安裝成功

dji@manifold2:~$ python3
Python 3.5.2 (default, Oct  8 2019, 13:06:37)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.cuda.is_available())
True

7、配置opencv3.4.1

參見我的另一篇博客： Caffe配置與編譯-----OpenCV-3.4.1配置及編譯

完成該步驟之後你將看到：

dji@manifold2:~$ pkg-config --modversion opencv
3.4.1
dji@manifold2:~$ python3
Python 3.5.2 (default, Oct  8 2019, 13:06:37)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import cv2
>>>

足夠的耐心和恆心，終能找到突破冰川的燦爛千陽！