TX2 jetpack4.2安装pytorch详细教程

TX2 刷完jetpack4.2固件后安装的ubuntu系统同时存在两个版本的python,一个是默认的python2.7,另一个是python3.6版本,我安装pytorch1.1.0使用的是2.7版本。

1、安装pytorch

(1)、将TX2开启最大工作模式

sudo nvpmodel -m 0 
cd usr/bin
sudo ./jetson_clocks 

(2)、安装pip

sudo apt-get install python-pip

pip install --upgrade pip    #更新pip版本

(3)、安装必要的依赖

sudo apt-get install libopenblas-base libopenmpi-dev

(4)、安装pytorch

TX2安装Pytorch最简单的方法就是下载nvidia官网提供的对应版本的pytorch的whl文件和教程进行安装,nvidia官方pythorch下载地址。本次选择的是安装的python2.7版本的pytorch1.1.0版本。

 

安装的方式有很多种:

第一种方法:通过wget下载whl文件有时会比较慢甚至下载失败,我是在windows环境下用利用KEXUE上网下载,速度会比较快,下载好后将文件放在home目录下然后执行后面2步。

wget https://nvidia.box.com/shared/static/1v2cc4ro6zvsbu0p8h6qcuaqco1qcsif.whl -O torch-1.1.0-cp27-cp27mu-linux_aarch64.whl
sudo apt-get install libopenblas-base libopenmpi-dev 
pip install future torch-1.1.0-cp27-cp27mu-linux_aarch64.whl

安装完成后可以在python环境下通过import torch进行验证

  

2 安装Torchvision v0.3.0 

通过以下命令即可完成安装

sudo apt-get install libjpeg-dev zlib1g-dev
#实测用手机热点下载会比较顺利
git clone --branch v0.3.0 https://github.com/pytorch/vision torchvision
cd torchvision
#这一步花费的时间较长
sudo python setup.py install
cd ../
pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

进行到sudo python setup.py install这一步时,报错如下: 

 

我也不知道啥原因我就编辑了bashrc文件加入了CUDA和CUDNN路径:

sudo gedit ~/.bashrc
export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu
export CUDNN_INCLUDE_DIR=/usr/include
export CUDA_ROOT="/usr/local/cuda-10.0/"
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64/:$LD_LIBRARY_PATH"
source ~/.bashrc

也可能是是安装pillow的过程失败。通过下面的语句可实现加速安装pillow,便可解决问题,然后接着执行后面的语句。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pillow==4.1.1

3 安装验证

通过一下命令进行安装验证

python -c "import torch ; printtorch.__version__)"

安装正确的话,结果如:

4 测试

新建文件

gedit test.py

内容如下:

import torch
printtorch.cuda.is_available))
a = torch.cuda.FloatTensor2)
printa)
b = torch.randn2).cuda)
printb)
c = a + b
printc)

执行结果:

python  test.py

输出:

  

Published by

风君子

独自遨游何稽首 揭天掀地慰生平

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注