Skip to main content

PyTorch环境搭建

PyTorch环境搭建1、系统信息2、安装依赖3、安装PyTorch3.1、离线安装3.2、在线安装4、验证安装参考资料

PyTorch 是一个由 Meta（原 Facebook）开发的开源深度学习框架，以灵活性和易用性著称，广泛用于学术研究和工业应用。

1、系统信息​

image-20250121150858152

2、安装依赖​

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 update sudo apt -y install libcusparselt0 libcusparselt-dev -y xxxxxxxxxx sudo apt remove python3-sympy

3、安装PyTorch​

PyTorch的安装选择离线和在线的其中一种方式即可！

3.1、离线安装​

手动进入下载网址下载指定的PyTorch版本。

下载网址：https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/

xxxxxxxxxx cd Downloads/ xxxxxxxxxx sudo pip3 install torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

3.2、在线安装​

xxxxxxxxxx export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v61/pytorch/torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl xxxxxxxxxx sudo python3 -m pip install --no-cache $TORCH_INSTALL

4、验证安装​

xxxxxxxxxx python3 -c "import torch; print(f'Torch: {torch.__version__}')"

image-20241226201901856

参考资料​

https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html

https://developer.nvidia.com/cusparselt-downloads