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