Jetson Orin NX 安装 anaconda、cuda、torch、torchvision

  第一次接触踩了不少坑,切忌不要按照常见服务器、电脑的思路安装。

安装 JetPack 套件

JetPack 是 Nvidia为 Jetson 系列开发板开发的一款软件开发包,常用的开发工具基本都有,安装 Jetson 会自动的将匹配版本的CUDA、cuDNN、TensorRT等安装好,极大地方便了开发者!几乎就是一键傻瓜安装,爽到爆!
在这里插入图片描述

sudo apt update
sudo apt install nvidia-jetpack
sudo apt show nvidia-jetpack

安装pytorch with cuda

安装完 JetPack 后会自动安装cuda,pytorch不能直接装pytorch官网的,需要修改与 JetPack 对应的版本,更多版本详见pytorch。下面圈红的是我安装的版本,我用的是 JetPack 5.1,cuda版本对应的是11.4,cuda版本可以通过nvcc -V查看。
在这里插入图片描述
点击箭头的位置开始下载.whl文件夹,下载后在.whl文件夹文件对应的文件夹内输入下面命令即可安装

<
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值