1.概述
Jetson平台上的Pytorch
Pytorch(适用于JetPack)是一个针对深度学习优化的张量库,能够调用 GPU 和 CPU。自动微分是通过基于磁带的系统在功能层和神经网络层级别完成的。此功能带来了高水平的灵活性,且具备深度学习框架的速度,并提供了类似 NumPy 的加速功能。这些 NVIDIA 提供的可再发行组件Python pip wheel 安装程序是适用于 PyTorch 的 ,且具有 GPU 加速和对 cuDNN 的支持。这些软件包需要安装在指定的 JetPack 版本。
2.对应版本列表
Torchvision对应Pytorch版本列表
3.安装说明
Pytorch安装方式:(以2.1版本为