谷歌colab中的MAGMA

Question

每当我使用安装了 magma-2.6.1 的 google colab 运行非常简单的代码时，我总是会收到以下错误“magma_getdevice_arch 中的错误：MAGMA 未初始化（首先调用 magma_init()）或设备损坏”。

////////////////////////

#include <iostream>
#include "cublas_v2.h"
#include "magma_v2.h"     
#include "magma_lapack.h" 

using namespace std;
int main ( int argc , char **argv ){

 magma_init();

magma_getdevice_arch(); 

magma_finalize();
return 0;
} 

Answer 1

这个问题似乎与 PyTorch 版本有关，对于 Colab Systems 上安装的 CUDA 版本来说太新了。

运行pip install --user torch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 torchtext==0.10.0降级。

答案摘自： https://www.kaggle.com/product-feedback/279990