Google Colab 中的 MAGMA
每当我使用安装了 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;
}
i always get the following error "Error in magma_getdevice_arch: MAGMA not initialized (call magma_init() first) or bad device" whenever i run a very simple code using google colab with magma-2.6.1 installed.
////////////////////////
#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;
}
如果你对这篇内容有疑问，欢迎到本站社区发帖提问 参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱，如果其他用户或者作者回复了您的评论，将不能在第一时间通知您！
发布评论
评论（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
It seems that this issue has to do with a version of PyTorch too new for the version of CUDA installed on Colab Systems.
Run
pip install --user torch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 torchtext==0.10.0to downgrade.Answer taken from: https://www.kaggle.com/product-feedback/279990