在国产超算平台上(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/

 

posted on 2023-07-07 08:13  Angry_Panda  阅读(872)  评论(0)    收藏  举报

导航