在国产超算平台上(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 阅读(872) 评论(0) 收藏 举报

浙公网安备 33010602011771号