最近在国产超算平台上安装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/

