gpu - 谷歌 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

似乎这个问题与安装在 Colab Systems 上的 CUDA 版本相比，PyTorch 版本太新有关。

运行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