Jetson NX 配置pytorch
参考文档
备注:参考博客最后安装的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

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

被折叠的 条评论
为什么被折叠?



