【Jetson安装PyTorch&Torchvision极简方式】

最新推荐文章于 2025-04-15 16:28:42 发布

一只云卷云舒

最新推荐文章于 2025-04-15 16:28:42 发布

阅读量425

点赞数 7

CC 4.0 BY-SA版权

文章标签： pytorch 人工智能 Torchvision

本文链接：https://blog.youkuaiyun.com/python_yjys/article/details/145451271

1. JetPack 5.1.2

安装 torch 2.1.0 和 torchvision 0.16.2

pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.16.2+c6f3977-cp38-cp38-linux_aarch64.whl

2. JetPack 6.1

安装 torch 2.5.0 和 torchvision 0.20

pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.5.0a0+872d972e41.nv24.08-cp310-cp310-linux_aarch64.whl
pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.20.0a0+afc54f7-cp310-cp310-linux_aarch64.whl

安装 cuSPARSELt 的依赖性问题 torch 2.5.0

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/arm64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install libcusparselt0 libcusparselt-dev