工作需要将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