Pytorch1.11.0+torchvision0.12.0(提取码:hdod)
一.使用Jeston平台:This guide provides instructions for installing PyTorch for Jetson Platform
NVIDIA Jetson AGX Orin 开发人员套件包括一个高性能、节能的 Jetson AGX Orin 模块,并且可以模拟其他 Jetson 模块。您现在拥有高达 275 TOPS 和 8 倍于 NVIDIA Jetson AGX Xavier 的性能,在同样紧凑的外形尺寸下可以开发先进的机器人和其他自主机器产品。
在为 Jetson 安装 PyTorch 之前,请确保:
1.在您的 Jetson 设备上安装 JetPack(Jetpack安装)
2.安装 PyTorch 所需的系统包:
sudo apt-get -y update
sudo apt-get -y install autoconf bc build-essential g++-8 gcc-8 clang-8 lld-8 gettext-base gfortran-8 iputils-ping libbz2-dev libc++-dev libcgal-dev libffi-dev libfreetype6-dev libhdf5-dev libjpeg-dev liblzma-dev libncurses5-dev libncursesw5-dev libpng-dev libreadline-dev libssl-dev libsqlite3-dev libxml2-dev libxslt-dev locales moreutils openssl python-openssl rsync scons python3-pip libopenblas-dev
二.Jetson AGX Orin安装Pytorch:PyTorch for Jetson
用jtop命令查看版本信息:
对应版本:Jetson AGX Orin+JetPack 5.0.1[L4T R34.1.1],于是下载:
- Python 3.8 - torch-1.11.0-cp38-cp38-linux_aarch64.whl 729(一般要翻)这里下载好保存在百度网盘里需要的可以下载使用(链接在最上面)
- torchvision下载网址(GitHub - pytorch/vision)选择main-->Tags
目前该网址上还没写Pytorchv1.11-torchvison0.x.x,在Pytorch官网上对应的是torchvision0.12.0
上面的准备工作做好后开始安装:
三.安装(在自己创建的虚拟环境下python=3.8)
1.将两个文件放在一个文件夹下,并把vision-0.12.0.zip提取到此处重命名为torchvison
2.在此处打开命令终端运行:
a.安装pytorch
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
pip3 install numpy torch-1.11.0-cp38-cp38-linux_aarch64.whl
b.安装torchvison
cd torchvision
export BUILD_VERSION=0.12.0
python3 setup.py install --user
cd ../
pip install 'pillow<7'
四.查看版本
python
import torch
import torchvision
torch.__version__
torchvision.__version__
五.遇到的问题
问题:
ModuleNotFoundError: No module named 'chardet'
ModuleNotFoundError: No module named 'charset_normalizer'
解决:pip install chardet