本文主要是介绍Jetson Orin Nano安裝torch torchvision,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
#https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048# 準備好2個文件放在Dir裏1. torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl 下載地址 https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048 注意JetPack版本,我使用的是5.1.32. vision-0.16.1.tar.gz 下載地址 https://github.com/pytorch/vision/releases
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev sudo apt-get install libjpeg-dev libpng-dev pip3 install 'Cython<3' pip3 install numpy export DIR=/home/$USER/Documents/torch2.1cd $DIR
pip3 install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64tar -zxvf vision-0.16.1.tar.gzcd vision-0.16.1python3 setup.py install --user
# 驗證 #cd ../ 必須要離開vision的目錄 #python3 #import torchvision #print(torchvision.__version__)# 版本對應 # PyTorch v1.0 - torchvision v0.2.2 # PyTorch v1.1 - torchvision v0.3.0 # PyTorch v1.2 - torchvision v0.4.0 # PyTorch v1.3 - torchvision v0.4.2 # PyTorch v1.4 - torchvision v0.5.0 # PyTorch v1.5 - torchvision v0.6.0 # PyTorch v1.6 - torchvision v0.7.0 # PyTorch v1.7 - torchvision v0.8.1 # PyTorch v1.8 - torchvision v0.9.0 # PyTorch v1.9 - torchvision v0.10.0 # PyTorch v1.10 - torchvision v0.11.1 # PyTorch v1.11 - torchvision v0.12.0 # PyTorch v1.12 - torchvision v0.13.0 # PyTorch v1.13 - torchvision v0.13.0 # PyTorch v1.14 - torchvision v0.14.1 # PyTorch v2.0 - torchvision v0.15.1 # PyTorch v2.1 - torchvision v0.16.1 # PyTorch v2.2 - torchvision v0.17.1 # PyTorch v2.3 - torchvision v0.18.0
这篇关于Jetson Orin Nano安裝torch torchvision的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!