当前位置: 首页 > article >正文

jetson agx orin 的pytorch、torchvision安装

首先使用conda创建一个虚拟环境,python=3.8,只能使用3.8,因为后面使用NVIDIA提供的torch包对py环境有要求。
使用

conda search cudatoolkit

查看自己支持的cuda版本,我选的11.4
然后进入英伟达的torch网站
pytorch英伟达下载地址

下载后

pip install torch-1.12.0a0+nnnnnnnnnnnn_linux_aarch64.whl

这个时候torch就安装完了,还需要安装torchvision
pytorch和torchvision版本对应
找到对应版本,torchvision需要自己下载编译

git clone --branch v0.16.0 https://github.com/pytorch/vision torchvision

下载自己对应版本的torchvision,然后在终端的虚拟环境输入下列命令

cd torchvision
export BUILD_VERSION=v0.16.0

在终端的虚拟环境输入下列命令

python3 setup.py install --user

编译完成即可。

参考:参考链接


http://www.kler.cn/news/360207.html

相关文章:

  • el-table表格数据处理,列表将变更前数据放置在前面,变更后数据放在表格后面
  • 第1篇:计算机网络概述与基础
  • 【小白学机器学习15】 概率论的世界观
  • Web,RESTful API 在微服务中的作用是什么?
  • 【修订中】ffmpeg 知识点
  • 从传统到智能,从被动监控到主动预警,解锁视频安防平台EasyCVR视频监控智能化升级的关键密钥
  • 【小程序】-基础语法(二)
  • Netty初体验-1-NIO基础补漏
  • 域渗透之 内网穿透 隧道技术聊聊看 环境搭建 regeorg技术 proxifier联合使用 一步步让你明白如何玩转隧道技术 含对应软件下载
  • Yolo目标检测:实时性与准确性的完美结合
  • 一文2500字从0到1实现压测自动化!
  • 宠物用品电商网站:SpringBoot框架设计与开发
  • 回忆Web编程的岁月变迁
  • Anchor DETR论文笔记
  • Docker私有仓库部署与管理
  • 【含开题报告+文档+PPT+源码】基于SpringBoot电脑DIY装机教程网站的设计与实现
  • FairGuard游戏加固全面适配纯血鸿蒙NEXT
  • 机器人如何强化润滑与维护,减少关节磨损?
  • 如何通过 systemd 确保服务自动重启:以 java-test 服务为例
  • 计算机网络:数据链路层 —— 交换式以太网