Nvidia Jetson Nano配置深度学习环境 ARM构架安装torch

目录

1、安装pyenv并配置环境

1.1 安装 pyenv 所需的一些基础依赖

1.2 安装pyenv

1.3 安装python

1.4 安装 pyenv-virtualenv 插件

1.5 创建虚拟环境

2、在虚拟环境中安装安装pytorch和trochvision

2.1 安装pytorch

2.2 安装torchvisio

1、安装pyenv并配置环境

1.1 安装 pyenv 所需的一些基础依赖

sudo apt-get update
sudo apt-get install -y make build-essential libssl-dev zlib1g-dev libbz2-dev \
libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev xz-utils tk-dev \
libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git python-openssl

1.2 安装pyenv

安装命令：

​git clone GitHub - pyenv/pyenv: Simple Python version management ~/.pyenv

配置：安装完 pyenv 后，需要将 pyenv 目录添加到你的 PATH 环境变量中，以及配置一些额外的环境变量，以确保 pyenv 正常工作。你需要vim ~/.bashrc命令进入并将以下内容添加到你的 ~/.bashrc 文件中

export PYENV_ROOT="$HOME/.pyenv"
export PATH="$PYENV_ROOT/bin:$PATH"
eval "$(pyenv init --path)"
eval "$(pyenv virtualenv-init -)"

注意：esc退出  :wq保存·bashrc文件 

为了应用上述更改，你需要重新加载你的 shell 配置文件：source ~/.bashrc

安装完成后，可以通过命令验证 pyenv 是否正确安装：pyenv --version

1.3 安装python

# 查看可安装的python的版本
pyenv install –list  
# 安装python：（我这里是安装3.6.17的python版本，“-v”是显示安装的详细信息，也可以不加，这一步可能会很慢，还可能会报错，建议多次尝试或者换个版本安装）
pyenv install -v 3.6.17
# 查看安装情况
pyehv versions或者pyenv version
# 设置全局 Python 版本
pyenv global 3.6.17

1.4 安装 pyenv-virtualenv 插件

pyenv-virtualenv 是一个 pyenv 的插件，它简化了虚拟环境的管理。通过安装这个插件，你可以更方便地创建、激活和管理虚拟环境。

git clone https://github.com/pyenv/pyenv-virtualenv.git $(pyenv root)/plugins/pyenv-virtualenv

1.5 创建虚拟环境

使用 pyenv virtualenv 命令可以创建一个新的虚拟环境。例如，要在 Python 3.6.17 中创建一个名为 "myenv" 的虚拟环境，可以执行以下命令：

pyenv virtualenv 3.6.17 myenv

激活和退出虚拟环境：

pyenv activate myenv
pyenv deactivate

2、在虚拟环境中安装安装pytorch和trochvision

两个文件我在这里提供网盘链接：https://pan.baidu.com/s/16dRiPOaslWvmzkjSFiS5Xg 
提取码：desk 

下载完你会看到一个whl文件和torchvision文件，pytorch直接是一个安装包可以直接安装，torchvision是一个文件夹，需要手动安装。

2.1 安装pytorch

# 进入pyenv环境
pyenv activate myenv
# 进入到pytorch的whl文件所在目录的终端，执行安装命令：
pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

测试是否安装成功：

python  #进入python编码
import torch
print(torch.__version__)  # 看到torch的版本号，torch安装成功!!!
torch.cuda.is_available() # 输出为True，则说明能用

2.2 安装torchvision

# 进入trochvision文件夹中，执行命令：
export BUILD_VERSION=0.9.0
# 然后执行安装命令：
python setup.py install（需要比较长时间，如果有出现pillow版本报错问题，可以在setup.py文件里面修改版本）

测试trochvision是否安装成功：

python
import torchvision
print(torchvision.__version__)  # 看到torchvision版本号，成功！！！

参考文章：

Linux安装pyenv（python多版本）教程_[root@mydocker.io ~]# pyenv versions * system (set-CSDN博客

yolov5 部署jetson nano（通用） 保姆级教学_jetsonnano部署anaconda和yolov5-CSDN博客