PyTorch(用于JetPack)是一个优化的深度学习张量库,使用GPU和CPU。自动微分是通过基于磁带的系统在功能和神经网络层级别完成的。此功能带来了高度的灵活性,作为深度学习框架的速度,并提供类似NumPy的加速功能。这些 NVIDIA 提供的可再发行组件是 PyTorch 的 Python pip 轮安装程序,具有 GPU 加速和对 cuDNN 的支持。这些软件包旨在安装在指定版本的 JetPack 之上。
在安装 PyTorch for Jetson 之前,请确保:在您的 Jetson 设备上安装 JetPack。
安装 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++