PyTorch环境搭建
PyTorch环境搭建1、系统信息2、安装依赖3、安装PyTorch3.1、离线安装3.2、在线安装4、验证安装参考资料
PyTorch 是一个由 Meta（原 Facebook）开发的开源深度学习框架，以灵活性和易用性著称，广泛用于学术研究和工业应用。
1、系统信息
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__}')"
参考资料
https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html