请看这个博主写的链接,写的非常好,目前我已经安装成功了,不同的是我是在Anaconda虚拟环境中安装的。
原博客链接:https://blog.csdn.net/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__