jeston nano 安装pytorch

该博客指导如何在Jetson设备上安装PyTorch 1.8.0和torchvision 0.9.0。首先,通过官方或第三方链接下载适用于JetPack 4.4.x或4.5.x的Python 3.6的whl文件。然后,使用sudo apt-get安装必要的依赖,如pip和numpy。接着,使用pip3安装PyTorch whl文件。为安装torchvision,需要手动下载源码并使用python3 setup.py install进行安装。最后,通过Python脚本验证安装是否成功。

(1)主要参考

http://blog.openpilot.cc/archives/2271

(2)官网下载地址

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-8-0-now-available/72048

(3)注意自己的镜像对应的版本,这里推荐下载1.8版本

 PyTorch v1.8.0

 如果无法下载,进入如下好人的链接

https://download.youkuaiyun.com/download/qq_51095731/16753369

(3)安装方法

sudo apt-get install pytho
根据引用和的信息,Jetson Nano安装GPU版本的PyTorch有两种方法: 1. 自己编译PyTorch的源代码。 2. 下载NVIDIA官方或其他人预先编译好的PyTorch的轮子。 在Jetson Nano安装GPU版本的PyTorch需要注意它属于aarch64架构,不同于一般的x86_64的Linux系统架构。而PyTorch官网提供的aarch64架构的版本是CPU版本的,因此在Nano安装CPU版本的PyTorch会失去GPU的加速优势。不过,NVIDIA官网提供了GPU版本的PyTorch,目前只支持Python 3.6版本。 您可以在NVIDIA官方网站下载适用于Jetson Nano的GPU版本的PyTorch。具体的安装步骤可能会因为软件版本的变化而略有不同,建议您参考官方文档或者网上的教程进行安装。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [nvidia jetson nano如何安装cuda版的pytorch](https://blog.youkuaiyun.com/weixin_43945848/article/details/127122522)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *3* [Nvidia Jetson nano 安装Archiconda、gpu版torch、踩坑记录](https://blog.youkuaiyun.com/qq_41451125/article/details/116262611)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值