- 博客(2)
- 收藏
- 关注
原创 jetson系列pytorch和tensorflow安装方式
本文介绍了在Jetson设备上正确安装PyTorch以启用GPU加速的方法。由于Jetson的Tegra GPU架构特殊,不能直接使用x86平台的PyTorch安装包。作者详细说明了如何根据JetPack版本(如5.1.1)下载对应的PyTorch和TensorFlow的aarch64架构whl文件,并提供了官方下载链接示例。安装步骤包括:先安装匹配版本的PyTorch(如2.0.0),再安装对应的torchvision(如0.15.1)。最后通过验证CUDA可用性、版本号及GPU型号来确认安装成功。该方法
2025-11-25 11:48:37
132
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅