侧边栏壁纸
  • 累计撰写 132 篇文章
  • 累计创建 16 个标签
  • 累计收到 4 条评论

目 录CONTENT

文章目录

Jetson 安装 Pytorch 深度学习框架

NVIDIA Jetson是非常高性能的深度学习平台,可以高效地运行许多 PyTorch 深度学习模型,并且 NVIDIA 为其开发了高级的 Jetson Inference API。

设置 Jetson

sudo pip3 install jetson-stats
sudo jtop

可以看到Jetson平台的相关资源,还可以查看已安装的 CUDA 版本:

2025-04-13 13-16-47屏幕截图
2025-04-13 13-17-31屏幕截图

安装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
0

评论区