TX2入门教程软件篇-安装PyTorch(Caffe2)v1.2.0(jetpack4.2)

TX2入门教程软件篇-安装PyTorch(Caffe2)v1.2.0(jetpack4.2)

说明：

• 介绍如何在tx2 jetpack3.3安装PyTorch(Caffe2)v1.2.0

步骤：

• 新建目录

mkdir -p ~/dl/PyTorch
cd ~/dl/PyTorch

• 下载Python 2.7 pip wheel文件

wget https://nvidia.box.com/v/torch-1-2-cp27-jetson-jp421

• 下载Python 3.6 pip wheel文件

wget https://nvidia.box.com/v/torch-1-2-cp36-jetson-jp421

• 安装脚本：

# Python 2.7 (download pip wheel from above)
$ pip install torch-1.2.0a0+8554416-cp27-cp27mu-linux_aarch64.whl --user

# Python 3.6 (download pip wheel from above)
$ pip3 install numpy torch-1.2.0a0+8554416-cp36-cp36m-linux_aarch64.whl

测试:

参考：

• https://elinux.org/Jetson_Zoo#PyTorch_.28Caffe2.29