Jetson TX2安装编译pytorch

工作需要将deepsort移植到TX2上面，从网上搜集了教程。

安装教程：

1. 最好是重新安装一个jetpack系统，3.2及以上都可以。

2. 首先安装环境依赖：因为是在python3下面安装pytorch，所以使用pip3。

    sudo apt install libopenblas-dev libatlas-base-dev liblapack-dev

    # For OpenCV

    sudo apt install liblapacke-dev checkinstall

    sudo apt-get install python3-pip

    sudo apt-get install python3-dev

    # 安装python库

    sudo apt-get install python3-dev python3-pip libxml2-dev libxslt1-dev zlib1g-dev libffi-dev libssl-dev

    sudo apt-get install gfortran 

    sudo pip3 install numpy scipy

    sudo pip3 install pyyaml

    sudo pip3 install scikit-build

    sudo apt-get -y install cmake

    sudo pip3 install cffi

3. 添加CUDNN路径：我们刷机之后CUDNN和CUDA都是安装好的，但是我们需要手动将CUDNN添加到环境中。

    sudo gedit ~/.bashrc

    export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu

    export CUDNN_INCLUDE_DIR=/usr/include

    source ~/.bashrc

4. 下载pytorch安装包，并且安装pip3所需要的其他安装包。

    git clone http://github.com/pytorch/pytorch

    cd pytorch

    sudo pip3 install -U setuptools

    sudo pip3 install -r requirements.txt

5. 选择pytorch要编译的版本。

    # 这里选择1.0.0版本进行编译

    git checkout tags/v1.0.0 -b build  #这一句话很重要是选择你要编译的pytorch版本

    git submodule update --init --recursive

    这里需要注意的是当你执行git submodule update --init --recursive  这句命令时候需要你输入你的github的用户名和密码来下载 https://github.com/NervanaSystems/nervanagpu.git 这个安装包，但是这个安装包已经丢失下载不到了。所以我们需要将这个包在git上删除掉，然后再运行git submodule update --init --recursive。

    命令行如下：删除nervanagpu这个包。

    git clean -xdf

    git submodule deinit -f .

    git submodule sync

    git submodule update --init --recursive

6. 调整TX2的工作效率

    sudo nvpmodel -m 0        # 切换工作模式到最大

    cd  /usr/bin/

    sudo ./jetson_clocks      # 强制开启风扇最大转速

7. 开始编译pytorch：这两句命令编译时间很长一般在4小时左右。

    sudo python3 setup.py build

    ​

    NO_SYSTEM_NCCL=1 DEBUG=1 sudo -E python3 setup.py install

8. 后续操作

    sudo apt clean

    sudo apt-get install libjpeg-dev zlib1g-dev

    git clone https://github.com/python-pillow/Pillow.git

    cd Pillow/

    sudo python3 setup.py install

    sudo pip3 install pandas # ~20-30 min

    sudo pip3 install Cython

    sudo pip3 install scikit-image

    sudo pip3 --no-cache-dir install torchvision

9. 验证安装

    $ sudo python3  #运行python3的时候加上管理员权限，如果不加sudo，程序会报CUDA错误

    >>> import torch

    >>> 

————————————————

版权声明：本文为CSDN博主「donkey_1993」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。

原文链接：https://blog.csdn.net/donkey_1993/article/details/102544868