一文贯通jetson平台下源码安装Pytorch 从而使用分布式计算torch.distributed ReduceOp

已于 2024-05-17 16:53:05 修改
阅读量558 收藏 14
点赞数 9
分类专栏： jetson pytorch mmdet3d 环境搭建 文章标签： pytorch nvidia jetson
于 2024-05-16 15:49:22 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_38418182/article/details/138960848
版权

nvidia jetson 平台 源码安装 pytorch

1、 确认平台环境版本

查看jetpack版本

有个比较简单方便的方法，直接使用nvidia的 jtop，可以综合看到系统的各项信息。没有的可以参照这篇博文安装 jtop

sudo jtop

这里可以看到我的
jetpack版本：5.1.2
CUDA版本：11.4

2、确认适配的torch版本

torch与jetpack的版本对应要求很严格，版本不对会安装失败

可以去Nivida pytorch for jetson查找与jetpack版本对应的pytorch版本预编译wheel
在这里插入图片描述
我这里适配的版本是torch 2.1.0，正常情况下，下载官方提供的whl文件，执行：

pip3 install torch-2.1.0.....#前面下载的whl文件，名字太长我不打
最低0.47元/天 解锁文章
黑翅
关注
  • 9
    点赞
  • 踩
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
jetson-tx2-pytorch:在Nvidia Jetson TX1TX2上安装PyTorch
05-11
在NVIDIA Jetson TX1 / TX2上安装PyTorch 是一个新的深度学习框架，可以在Jetson TX1和TX2板上很好地运行。 它安装起来相对简单快捷。 与TensorFlow不同，它不需要外部交换分区即可在TX1上构建。 尽管TX2具有足够的32 GB eMMC，但TX1仅具有一半，而且由于JetPack，Ubuntu软件包和安装工件的限制，它很容易用完空间。 下面的清理部分列出了一些简化方法，此处的步骤倾向于极简主义。 PyTorch开发人员推荐Anaconda发行版。 我找不到用于ARM64的最新Anaconda设置，因此我使用了全局python库。 提示：在TX2上，运行~/jetson_clocks.sh加速CPU的运行并启用两个以上的内核。 这样可以将PyTorch的编译时间从45分钟减少到37分钟。 我没有在TX1上进行测试，但是可以预期的是速度不会
基于NVIDIA jetson平台Torch安装教程 +Jetson-Inference使用
09-06
基于NVIDIA jetson平台Torch安装教程 +Jetson-Inference使用， 老实说，这是我最喜欢的一部分了，最近沉迷于寻找各种用于不同阶段的加速工具，jetson平台的TensorRT属于“推理阶段加速”，也是比较实用的一种加速，之后找个时间我会专门写一节总结，介绍各种加速工具给你，这里就不多说了。 这本书中关于使用TensorRT的部分比较坑，有很多东西我没有顺利安装。这里总结一下我最后成功方法。
Jetson设备安装Pytorch和Torchvision
最新发布
tw_fae的博客
06-03 241
详细介绍了Jetson设备不同版本系统安装pytorch的安装方式，方便大家及时开展工作。
NVIDIA Jetson系列安装torch, mmdetection+mmaction2的环境配置全流程，并成功部署open-mmlab的中时空动作检测的项目
AI小白炼金术师
01-11 3815
NVIDIA Jetson系列安装torch, mmdetection+mmaction2的环境配置全流程 ，并部署open-mmlab的项目(mmdetection+mmaction2)，并解决预见到的问题
jetson nano装torch和torchvision
weixin_45235219的博客
01-29 2322
学习jentson nano装torch和torchvision时的一些随笔
jetson Xavier nx安装torch和torchvision，并解决解决版本不匹配（报错RuntimeError: Couldn‘t load custom C++ ops）的问题
hhfhng的博客
11-22 6640
首先，torch和torchvision都不能直接pip安装，以下的演示是以下的演示是torch1.10.0 + torchvision v1.11.1
JETSON TX2 NX配置conda、cuda、torch、torchvision环境保姆教程
weixin_72812064的博客
05-11 3927
上一期已经装好了系统本期配置一下环境sudo gedit ~/.bashrcexport CUDA_HOME=/usr/local/cuda-10.2 export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda-10.2/bin:$PATHnvcc -V为JETSON TX2 NX的架构是aarch64,与win10,linxu不同,所以不能安装Anaconda可以安装一个替代它
jetson xavier下编译的pytorch 1.10.0安装包，基于jetpack5.0.1
06-09
jetson-version-1-11-now-available/72048）只提供了jetpack5.0下的pytorch 1.11和1.12版本，但是由于我要装的其他包不支持这么高版本的pytorch，所以自己根据上面网站里的教程从源码编译了pytorch 1.10.0，...
jetson nano安装pytorch
11-16
在nvidia的jetson nano、AGX、TX2等开发板上安装pytorch环境。资源包括：torch-1.7.0-cp36-cp36m-linux_aarch64.whl以及1.6.0、1.5.0、1.4.0资源，torchvision，Miniforge-pypy3-4.9.0-3-Linux-aarch64.sh。
cuda 极简基于NVIDIA jetson平台Torch安装教程Jetson-Inference使用.zip
03-10
【标题】：“CUDA极简基于NVIDIA Jetson平台Torch安装教程及Jetson-Inference使用” 在NVIDIA的Jetson系列开发板上进行深度学习应用，CUDA和Torch的安装是必不可少的步骤。CUDA（Compute Unified Device ...
基于NVIDIA jetson平台Torch安装教程 +Jetson-Inference使用.zip
03-10
基于NVIDIA jetson平台Torch安装教程 +Jetson-Inference使用.zip
Jetson AGX Xavier安装torch、torchvision
zhangtaozg1的博客
05-04 889
进入pycharm新建环境的目录，把.whl放在这个目录下，执行pip install torch-1.11.0-cp38-cp38-linux_aarch64.whl。在命令行输入jtop，查看jetpack版本，注意下载对应版本的torch。2、在pycharm中安装需要的各个插件。5、下载并安装torchvision。3、安装离线torch，下载安装包。环境ubuntu20.4。1、安装pycharm。
Jetson nano安装torch和torchvision
mxh3600的博客
08-25 3328
Jetson nano安装torch和torchvision
Jetson Xavier NX安装pytorch深度学习环境
caiqidong321的博客
07-20 1955
本贴记录在Jetson Xavier NX开发板上部署pytorch GPU深度学习环境，其他型号的开发板步骤也类似
Jetson Nano目标检测手把手实战教程（pytorch训练、tensorrt推理，含完整代码和数据）
热门推荐
钱彬的博客
10-08 3万+
一、概述 1.1深度学习和边缘计算 2012年AlexNet模型在ImageNet数据集上取得了巨大成功，从此，深度学习技术开始进入黄金发展期。随后，深度学习在多个应用领域均大幅提升了性能指标，尤其在计算机视觉领域，超越SVM等一众传统机器学习方法。然而，深度学习的高准确性是以深度学习对计算和内存的高需求为代价的。由于输入数据可能具有很高的维数，并且需要对输入数据执行数百万次计算，因此训练和推理一个深度学习模型对空间和计算需求都很高。高准确性和高资源消耗是深度学习的特点。 ​ 为了满足深度学习...
在 NVIDIA Jetson上安装GPU版本的pytorch和torchvision
NotName_9758的博客
02-01 786
这些pytorch包仅支持python3.6，使用conda直接创建python3.6时会报错，修改命令为。参考链接【3】处下载torch安装包。来正确创建python3.6环境。
jetson的docker中装torch、torchvision
啊呜兮兮的博客
04-14 2474
极简： 安装 torch 命令 $ pip3 install torch -f https://torch.kmtea.eu/whl/stable-cn.html 安装 torchvision 命令 $ pip3 install torchvision -f https://torch.kmtea.eu/whl/stable-cn.html 详细错误经历∠(°ゝ°) 时间：2022/4/14 环境 设 备：联想 EA-B310 核心模块：NVIDIA Jetson Xavier NX 系
jetson安装torch和torchvision
qq_44235661的博客
10-29 207
jetson安装的torch和torchvision
jetson nx cuda 10.2 使用 gpu 版 torch
weixin_62589845的博客
03-30 294
gpu版本根据我查的资料，只能在python3.6下安装。或者在github上下载0.11.1版本压缩包到本地。下载下面网站中对应的torch whl文件到本地。然后cd到解压完的vison文件夹中。使用pip3安装whl。把这两个放到文件夹中。
jetson agx orin 安装pytorch
07-28
对于Jetson AGX Orin，可以按照以下步骤安装PyTorch： 1. 打开Jetson AGX Orin的终端。 2. 确保你的Jetson AGX Orin已经连接到互联网。 3. 运行以下命令来安装PyTorch的依赖项： ``` $ sudo apt-get update $ sudo apt-get install python3-pip libopenblas-base libopenmpi-dev ``` 4. 安装PyTorch。你可以通过在PyTorch官方网站（https://pytorch.org）上找到适用于Jetson AGX Orin的正确版本。 例如，如果你的Jetson AGX Orin运行的是Ubuntu 18.04，Python 3.6和CUDA 10.2，你可以运行以下命令来安装PyTorch： ``` $ wget https://nvidia.box.com/shared/static/wa34qwrwtk9njtyarwt5nvo6imenfy26.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl $ sudo apt-get install python3-pip libopenblas-base libopenmpi-dev $ sudo pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl ``` 请注意，上述链接是示例链接，你需要根据你的系统配置选择正确的链接。 5. 安装完成后，可以通过导入PyTorch来验证安装是否成功： ``` $ python3 >>> import torch >>> print(torch.__version__) ``` 如果没有出现错误并且成功打印了PyTorch的版本号，那么说明PyTorch已成功安装在你的Jetson AGX Orin上。 希望这些步骤对你有所帮助！如果需要进一步的帮助，请随时提问。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值