已有网站可以做服务器吗,云南网站开发网络公司前10,营销型网站是什么,深圳施工图制作想在Jetson AGX Orin创建一个虚拟环境#xff0c;然后安装pytorch#xff0c;过程中遇到了很多的坑#xff0c;这篇文章主要用于记录过程~因为Orin本身是Arm架构#xff0c;X86架构可以装Anaconda#xff0c;对于ARM要装archiconda。
1.安装archiconda
1.1确定操作系统架…想在Jetson AGX Orin创建一个虚拟环境然后安装pytorch过程中遇到了很多的坑这篇文章主要用于记录过程~因为Orin本身是Arm架构X86架构可以装Anaconda对于ARM要装archiconda。
1.安装archiconda
1.1确定操作系统架构
安装anaconda前需要确定Ubuntu20.04操作系统的架构
Ubuntu版本信息
$ lsb_release -a
No LSB modules are available.
Distributor ID: Ubuntu
Description: Ubuntu 20.04.3 LTS
Release: 20.04
Codename: focal查看操作系统架构
// 查询处理器位数当前 Linux 是 64 位则输出 amd64是 32 位则会输出 i386。
$ dpkg --print-architecture
amd64
// 查询架构64位ARM架构会输出aarch64
$ arch
aarch64Jetson AGX Orin的架构是aarch64和win10linux不同不能安装anaconda。所以安装archiconda作为替代
1.2下载archiconda
下载命令如下
wget https://github.com/Archiconda/build-tools/releases/download/0.2.3/Archiconda3-0.2.3-Linux-aarch64.sh1.3下载后执行命令
//给文件添加可执行权限
chmod x Archiconda3-0.2.3-Linux-aarch64.sh
./Archiconda3-0.2.3-Linux-aarch64.sh或者bash Archiconda3-0.2.3-Linux-aarch64.sh1.4配置环境变量
sudo gedit ~/.bashrc在~/.bashrc的最后一行添加
export PATH~/archiconda3/bin:$PATH~/archiconda3/bin是安装后conda文件的路径
1.5换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/2.安装Pytorch
安装前提需要有jetpack和CUDA11.4
2.1查看jetpack版本
sudo apt-cache show nvidia-jetpack得到的结果
Version:5.1.1-b56
Architecture:arm642.2查看CUDA版本
查看CUDA版本命令
nvcc -V出现报错
bash:nvcc:未找到命令解决
cd ~
sudo gedit ~/.bashrc在.bashrc文件最后添加以下代码
export PATH/usr/local/cuda-11.4/bin:$PATH
export LD_LIBRARY_PATH/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
export CUDA_ROOT/usr/local/cuda-11.4然后编译.bashrc文件
source ~/.bashrc2.3安装torch 1.13.0 GPU版本
2.3.1下载PyTorch预编译包
在Jetson上安装PyTorch官方推荐使用anaconda安装但是anaconda repository里没有aarch64架构的PyTorch预编译包
解决nvidia官网有编译过的二进制包在以下网站按步骤安装即可
https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048
根据自己的环境下载对应的版本这里是JetPack5.1.1python3.8虚拟环境 2.3.2进入虚拟环境
#xxx是python3.8的虚拟环境名
conda activate xxx2.3.3安装依赖
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-devpip install Cython
pip install numpy2.3.4安装PyTorch
chmod x torch-1.13.0a0d0d6b1f2.nv22.10-cp38-cp38-linux_aarch64.whl
pip install torch-1.13.0a0d0d6b1f2.nv22.10-cp38-cp38-linux_aarch64.whl2.4安装torchvision
选择和pytorch对应的torchvision版本pytorch1.13与torchvision0.14对应
2.4.1下载torchvision
torchvision下载网址 https://github.com/pytorch/vision
在main的Tag里面找到对应的0.14版本并下载解压
2.4.2安装torchvision
cd torchvision
export BUILD_VERSION0.14.0
python3 setup.py install --user
cd ../
pip install pillow72.4.3查看torch和torchvision版本
输入以下命令查看
python
import torch
import torchvision
torch.__version__
torchvision.__version__参考链接http://t.csdnimg.cn/UxiS6
http://t.csdnimg.cn/gkOKh
http://t.csdnimg.cn/vzGk0
