[orin] nvidia orin 上安装 pytorch 和 torchvision 实操

最新推荐文章于 2025-06-04 13:45:02 发布

原创最新推荐文章于 2025-06-04 13:45:02 发布 · 4.5k 阅读

CC 4.0 BY-SA版权

文章标签：

5 篇文章

订阅专栏

本文详细记录了在Orin设备上使用Anaconda虚拟环境安装Python 3.8、PyTorch 1.11.0及torchvision 0.12.0的过程。包括创建conda环境、安装系统依赖、PyTorch与torchvision的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

请看这个博主写的链接，写的非常好，目前我已经安装成功了，不同的是我是在Anaconda虚拟环境中安装的。
原博客链接：https://blog.youkuaiyun.com/beautifulback/article/details/125717717

这次我要安装的软件版本是:python3.8 + torch1.11.0+ torchvision0.12.0

首先说明，我拿到这个orin设备之前，已经有很多人在上面配置了，所以可能前人已经安装了很多依赖。所以本次安装有可能存在有些依赖没有安装的情况。

安装过程

1.首先创建conda环境

conda create -n py38 -y

2.然后安装依赖：

sudo apt-get install  libopenblas-base libopenmpi-dev libomp-dev

3.安装Pytorch

# 先进入虚拟环境
conda activate py38
# 安装pip依赖
pip install Cython numpy
# 安装Pytorch
pip install  torch-1.11.0-cp38-cp38-linux_aarch64.whl

4.安装torchvision
解压压缩包vision，然后进入vision文件夹内，然后输入以下指令安装

export BUILD_VERSION=0.12.0  
python3 setup.py install --user

这一步会比较慢，耐心等待，如果有除了torchvision以外的包没有安装上，那么可以再额外进行安装即可，注意提示上的版本要求，例如下面的pillow版本要求不大于7。

安装pillow<7

pip install 'pillow<7'

5.查看版本

python

# 输入
import torch
import torchvision
torch.__version__
torchvision.__version__