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 ; print(torch.__version__)"
安装正确的话,结果如:
4 测试
新建文件
gedit test.py
内容如下:
import torch print(torch.cuda.is_available()) a = torch.cuda.FloatTensor(2) print(a) b = torch.randn(2).cuda() print(b) c = a + b print(c)
执行结果:
python test.py
输出: