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

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

纠错,疑问,交流: 请进入讨论区点击加入Q群

获取最新文章: 扫一扫右上角的二维码加入“创客智造”公众号


标签: tx2入门教程软件篇