Jetson NX 配置 pytorch

原创

已于 2024-07-23 18:07:56 修改 · 811 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #深度学习

于 2023-03-20 13:25:35 首次发布

本文介绍了在JetsonNX上配置PyTorch和torchvision的完整流程，包括使用JetPack5.0.2和Ubuntu20.04的环境，安装miniconda，设置环境变量，以及安装依赖库。在安装torchvision时遇到的问题和解决方法也进行了说明，特别是处理了与PyTorch版本不兼容的情况。此外，还提到了针对报错ImportError:cannotimportname‘PILLOW_VERSION’from‘PIL’的解决方法。

Jetson NX 配置pytorch

参考文档

官方教程–还得是官方啊
参考博客1
参考博客2
参考博客3：NVIDIA JETSON XAVIER NX （四）安装Pytorch和torchvision
官方提供的离线包
pytorch/vision

备注：参考博客最后安装的pytorch无法使用cuda-----目前没找到原因，不知如何解决。
但是官方的教程没问题！

环境情况：

系统JetPack 5.0.2 --ubuntu20.04

配置流程

安装miniconda（流程省略）
conda create -n pytorch python=3.8

安装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;
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
conda activate pytorch
pip3 install Cython
添加环境变量

export PATH=/usr/local/cuda/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda