Jetson TX2安装pytorch 1.2 教程（Ubuntu版本18，cuda10.0）

最新推荐文章于 2026-04-10 11:35:27 发布

原创最新推荐文章于 2026-04-10 11:35:27 发布 · 1.6k 阅读

23 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#pytorch #ubuntu #深度学习

Jeston TX2 专栏收录该内容

2 篇文章

订阅专栏

本文详细记录了在JetsonTX2上安装pytorch1.2.0和torchvision0.4.0的过程，包括下载对应版本的whl文件，解决cuda兼容性问题，以及安装过程中遇到的网络连接难题。作者强调了针对不同硬件平台选择合适软件版本的重要性，并感谢提供帮助的在线资源。

Jetson TX2安装pytorch 1.2 教程（Ubuntu版本18，cuda10.0）

前言：这段时间自己做毕业设计，是用tx2加一个摄像头部署个神经网络模型。然后呢，我遇到了很多困难，这一个星期里改环境配置把我搞得想鼠。TX2上ubuntu的环境和大部分ubuntu版本是有区别的，TX2是arm类型的芯片，大部分的是x86类型的芯片。所以给TX2装一些包会很麻烦的，有些包不是架构体系不对应，就是版本低了。

我的tx2上已有的环境：

QT5.9.5
然后cuda版本是 cuda10.0
opencv是3.4.0

至于我毕设已经在网上kaggle和Colab这样的提供gpu算力的平台跑出来了一个.pt格式的文件。然后我在这台TX2上跑，发现自己缺了torch包，而后我上网搜“如何在Ubuntu18 环境下配置torch C++调用”之类的关键词，之后跟着一些教程装了pytorch官网最新的包，

pytorch官网 LibTorch的安装
和我自己的cuda版本不匹配阿
装了之后发现自己的cuda版本低，后来又去配自己的cuda，这才发现TX2的cuda应该是要自己通过刷机刷上去的，这就算了，刷机好像也没几个讲到cuda刷到11版本以上的（——坑蝶阿！！）。
好吧我去下pytorch版本低的总行了吧，于是这就是本文中心来了。

先贴上参考网址：

在Jetson TX2上傻瓜式安装pytorch（这个网址顾名思义：有很多简单直白有效的信息）

JETSON TX2 安装 CUDA9.0 及 CUDNN7 超详细（真实亲测）（这个网址是帮助那些没有在TX2上刷机过的小朋友们的，至于刷过机但还想刷的也可以看看，我在本文中不会再提到这个了）。

Jetson TX2安装pytorch 1.4 (jetpack4.3)详细教程（我本人主要模仿这位大佬的教程，配置符合自己cuda版本的pytorch）

主文：

根据NVIDIA官网的安装教程，网址：nvidia官方。（我想了想，毕竟老东家出的这个TX2嘛，他不出教程谁还能出教程阿。）TX2安装Pytorch最简单的方法就是下载nvidia官网提供的对应版本的pytorch的whl文件和教程进行安装。
本次选择的是安装的python 3.6版本的pytorch1.4.0版本。
在这里插入图片描述

安装torch

通过wget下载whl文件有时会比较慢甚至下载失败，我是在windows环境下用利用科学上网下载，速度会比较快，Python 3.6 - torch-1.2.0a0+8554416-cp36-cp36m-linux_aarch64.whl，我也是点击了这个连接（我有梯子），然后它网页自动给我下载好了，我再用U盘拷到tx2上去的。
下载好后将文件放在home目录下然后执行后面3步（代码里面下载的文件要根据自己的改哦）。

wget https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl -O torch-1.2.0a0+8554416-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.2.0a0+8554416-cp36-cp36m-linux_aarch64.whl

安装完成后可以在python环境下通过import torch进行验证，我的验证好了：
在这里插入图片描述

安装Torchvision v0.4.0

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

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

git clone --branch v0.4.0 https://github.com/pytorch/vision torchvision（这一步可把我折腾死了，github总是等不上去，逼得我又去找linux的梯子，百无聊赖之余，它竟然又连接成功了，然后这条卡了我半个小时的安装总算ok了。这一行还是一把辛酸泪阿。）

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

python3 -c "import torch ; print(torch.__version__)"

在这里插入图片描述

总结：

TX2的资料国内其实很少，在找资料的过程中，还是得先根据自己的硬件平台来找，否则路会越走越偏。感谢以上的参考连接，谢谢你们的付出，没有你们的付出我可能还要卡上很久的时间。

接下来我准备配置这个torch包，用C++调用它，请大家祝我成功吧。希望再把这个写出来吧。
真诚的希望帮到大家。