jetson agx orin 的pytorch、torchvision安装

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

使用

bash 复制代码
conda search cudatoolkit

查看自己支持的cuda版本,我选的11.4

然后进入英伟达的torch网站
pytorch英伟达下载地址

下载后

bash 复制代码
pip install torch-1.12.0a0+nnnnnnnnnnnn_linux_aarch64.whl

这个时候torch就安装完了,还需要安装torchvision

找到对应版本,torchvision需要自己下载编译

bash 复制代码
git clone --branch v0.16.0 https://github.com/pytorch/vision torchvision

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

bash 复制代码
cd torchvision
export BUILD_VERSION=v0.16.0

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

bash 复制代码
python3 setup.py install --user

编译完成即可。

参考:参考链接

相关推荐
九亿AI算法优化工作室&4 分钟前
结合大语言模型的机械臂抓取操作学习
人工智能·学习·语言模型·自然语言处理
kaamelai4 分钟前
Kaamel视角下的MCP安全最佳实践
大数据·人工智能·安全
我要学脑机8 分钟前
基于常微分方程的神经网络(Neural ODE)
人工智能·深度学习·神经网络
有颜有货19 分钟前
2025汽车制造企业数字化转型路径参考
人工智能·汽车·制造·数字化转型
阿星AI工作室22 分钟前
小白也能用AI开发「小红书自动归档多维表格」采集神器,躺平整理笔记真香
人工智能
云天徽上23 分钟前
【数据可视化-42】杂货库存数据集可视化分析
人工智能·机器学习·信息可视化·数据挖掘·数据分析
大模型真好玩26 分钟前
初学者必看大模型微调指南:Unsloth官方微调技巧大公开!
人工智能·python
自由随风飘34 分钟前
机器学习第三篇 模型评估(交叉验证)
人工智能·机器学习
vocal34 分钟前
谷歌第七版Prompt Engineering—第三部分
人工智能·后端
谈不譚网安40 分钟前
初识Python
开发语言·python