在国产超算平台上（aarch64架构）安装pytorch-cuda失败，究其原因竟是官方未提供对应的cuda版本——pip方式和conda方式均无法获得相应cuda版本
最近在国产超算平台上安装pytorch，但是怎么弄都会报错：
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
不论是使用pip还是conda都是无法再aarch64平台上安装pytorch的cuda版本，即使是使用conda方式依旧会报错，一开始总认为是自己安装方式不对，后来才发现其实pytorch官方根本就没有提供aarch64架构平台下的cuda版本。
给出pytorch官方给出的版本：
https://download.pytorch.org/whl/torch/
我们可以清楚的看到所有aarch64架构下的pytorch均只提供cpu版本而不提供gpu版本，而X86架构下的pytorch则各个版本均有：
https://download.pytorch.org/whl/torch/
本博客是博主个人学习时的一些记录，不保证是为原创，个别文章加入了转载的源地址，还有个别文章是汇总网上多份资料所成，在这之中也必有疏漏未加标注处，如有侵权请与博主联系。
如果未特殊标注则为原创，遵循 CC 4.0 BY-SA 版权协议。
posted on 2023-07-07 08:13 Angry_Panda 阅读(662) 评论(0) 编辑 收藏 举报
【推荐】国内首个AI IDE，深度理解中文开发场景，立即下载体验Trae
【推荐】编程新体验，更懂你的AI，立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包，你的智能百科全书，全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell：AI 加持，快人一步
· Manus重磅发布：全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后，我竟然真的恢复了删除的微信聊天记录！
· 没有Manus邀请码？试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
2021-07-07 （续） MindSpore 如何实现一个线性回归 —— Demo示例
2021-07-07 mindspore.ops.Pow()等算子不能处理float64类型的数据
2021-07-07 【转载】 图解协程调度模型-GMP模型
2020-07-07 两个 RTX2070 super 显卡 可不可以通过 nvlink交换机 进行P2P通信 呢？ （答案： 可以）
2020-07-07 （ Ubuntu系统下 ） vim插件安装 supertab 实现tab键的补全功能 （vim+python环境下）
2020-07-07 （ Ubuntu环境下 ）Vim插件推荐-Python自动补齐Vim插件jedi-vim的安装（使用插件管理器vundle进行安装）
2020-07-07 ( Ubuntu环境下 ) Vim插件管理器 Vundle 的安装