[英]MAGMA in google colab
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.每当我使用安装了 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;
}
It seems that this issue has to do with a version of PyTorch too new for the version of CUDA installed on Colab Systems.这个问题似乎与 PyTorch 版本有关,对于 Colab Systems 上安装的 CUDA 版本来说太新了。
Run pip install --user torch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 torchtext==0.10.0
to downgrade.运行
pip install --user torch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 torchtext==0.10.0
降级。
Answer taken from: https://www.kaggle.com/product-feedback/279990答案摘自: https://www.kaggle.com/product-feedback/279990
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.