NVIDIA Jetson是非常高性能的深度学习平台,可以高效地运行许多 PyTorch 深度学习模型,并且 NVIDIA 为其开发了高级的 Jetson Inference API。
设置 Jetson
sudo pip3 install jetson-stats
sudo jtop
可以看到Jetson平台的相关资源,还可以查看已安装的 CUDA 版本:
安装pytorch
一般推荐在conda环境中安装:
conda create -n torch python=3.8
conda activate torch
在 官网 根据 Jtop 的 Jetpack 和 pytorch 版本选择下载安装:
pip install torch-2.0.0-cp39-cp39-linux_aarch64.whl
在根据 pytorch 的版本安装 torchvision 的版本:
pip install --no-deps torchvision==0.15.0
评论区