记录安装GPU版本pytorch

本文讲述了作者在升级PyTorch版本遇到的问题,包括因缺失包而重新安装CUDA10.2,通过PyTorch官网找到对应版本,从下载地址下载.whl文件,在虚拟环境中使用pip进行安装,并验证安装成功的经历。
摘要由CSDN通过智能技术生成

最近跑代码之前的pytorch版本不能用,需要安装1.10以上的,记录一下。

1、我的cuda目录下缺少了很多包,所以重新安装了cuda10.2。

去cuda官网按照步骤一步一步安装就好。

2、到pytorch官网查看适合自己的pytorch 版本

https://pytorch.org/

比如我的版本是cuda10.2

直接在命令行安装我的是显示报错,找不到包,所以我要按照上面的版本一个一个下载到本地

3、下载网址:download.pytorch.org/whl/

4、下载之后在终端输入打开虚拟环境

5、pip  install  +下载的文件所在路径。这样一个一个下载就能安装好了。

6、验证一下

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
根据引用和的信息,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.csdn.net/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.csdn.net/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 ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值