TX2上安装pytorch

浅浅的花香味﹌ 2024-04-18 10:57 60阅读 0赞

由于TX2是arm架构，无法直接通过官方的方法——pip或conda等安装，需要自行编译github上的软件包才能使用。

注：本文略过TX2上的CUDA，Cudnn以及opencv的安装，这些部分的安装大家可以自行百度。

现在正式开始编译pytorch吧：

**1.打开终端，输入以下命令下载git包**

git clone https://github.com/pytorch/pytorch.git

![在这里插入图片描述][20190905181733391.png]  
下载完后，大家会看到在home下有了pytorch文件夹。

**2.切分版本**

cd到home下的pytorch文件夹，进行版本切分，我要使用pytorch0.3.1版本，所以依次输入以下命令进行切分：

cd '/home/nvidia/pytorch'
     
    git checkout -b v0.3.1 v0.3.1

**3.更新submodule**

此步不进行，在install的时候会报Could not find /home/nvidia/pytorch/torch/lib/gloo/CMakeLists.txt错误

在pytorch文件夹路径下，在终端输入以下命令进行更新：

git submodule update --init

**4.正式编译安装**

还是在当前pytorch文件夹路径下，输入以下命令进行编译安装：

sudo python setup.py install

切记一定要带上sudo，不然编译很久很久要完成后，你会发现报Permission denied的错误。

（编译时间很长，大家请耐心等待，睡一觉就好了）

**5.测试**

在python下输入import torch，没有显示错误则说明编译安装成功，如下图：  
![在这里插入图片描述][20190905181916519.png]

## 查看pytorch版本的方法 ##

import torch
    print(torch.__version__)  #注意是双下划线

[20190905181733391.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/f81358e670a44b239b1da4d729e32a78.png
[20190905181916519.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/f7158bb40f7c4422af1af21d45f84bba.png

发表评论取消回复

表情：

评论列表（有 0 条评论，60人围观）

还没有评论，来说两句吧...

相关阅读

相关 TX2上安装pytorch

于TX2是arm架构，无法直接通过官方的方法——pip或conda等安装，需要自行编译github上的软件包才能使用。注：本文略过TX2上的CUDA，Cudnn以及o...

浅浅的花香味﹌/ 2024年04月18日 10:57/ 0 赞/ 61 阅读

相关【玩转Jetson TX2 NX】（十）TX2 NX 安装Archiconda3+创建pytorch环境（详细教程+错误解决）

1 写在前面的话该篇文章创建pytorch环境部分，`大家谨慎参考`。本以为已经完全解决错误，但实际上pytorch最终并未配置在所创建虚拟环境中，本人能力有限，实在搞

水深无声/ 2023年09月25日 21:16/ 0 赞/ 55 阅读

相关 tx2安装时候 Drivers for Jetson TX2 & TX2i : E: Could not get lock /var/lib/apt/lists/lock - open

[https://askubuntu.com/questions/1109982/e-could-not-get-lock-var-lib-dpkg-lock-frontend

- 日理万妓/ 2023年06月23日 14:28/ 0 赞/ 9 阅读

相关 TX2刷机与相关软件安装

文章目录一、刷机准备二、JetPack4.2刷机过程三、安装 qt creator 四、安装 Tensorfl

叁歲伎倆/ 2022年12月29日 14:23/ 0 赞/ 213 阅读

相关 Jetson TX2-JetPack 4.3 安装 pytorch和torchvision

声明： TX2系统： `Ubuntu18.04` python版本：`3.6.9` 刷入`JetPack 4.3`，`apt`更换为国内源。选择这个版本的原因是对应`c

浅浅的花香味﹌/ 2022年11月17日 16:09/ 0 赞/ 188 阅读

相关 TX2

重装系统： [https://blog.csdn.net/qq\_41883764/article/details/89164860][https_blog.csdn.ne

不念不忘少年蓝@/ 2022年11月12日 08:22/ 0 赞/ 153 阅读

相关英伟达TX2学习笔记——tx2简介

一、相关资料搜集官方环境搭建和入门例程： [https://www.yahboom.com/study/Jetson-TX2][https_www.yahboom.

傷城~/ 2022年11月07日 15:55/ 0 赞/ 302 阅读

相关 python2.7安装pytorch_PyTorch安装

这个系列写了好几篇文章，这是相关文章的索引，仅供参考：以下是相关深度学习工具包的安装，包括Tensorflow, PyTorch, Torch等：首先安装libcupt

约定不等于承诺〃/ 2022年10月22日 12:51/ 0 赞/ 336 阅读

相关在TX2上安装一些必要的库文件

1. liblas sudo aptitude install liblas-c-dev liblas-bin liblas-dev liblas-c3 pytho

╰+哭是因爲堅強的太久メ/ 2022年10月16日 09:53/ 0 赞/ 110 阅读

相关 Jetson TX1/TX2配置教程--拷贝离线安装包

使用JetPack 2.3配置TX1 / 使用JetPack 3.0配置TX2 配置JetPack 2.3/JetPack 3.0 JetPack 系列是NVIDIA发

雨点打透心脏的1/2处/ 2022年06月05日 03:36/ 0 赞/ 299 阅读