CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`
时间: 2023-09-21 14:11:08 浏览: 251
CUBLAS_Library_cublas_CUDA线性代数函数库文档_版本10.2_
5星 · 资源好评率100%
这个错误通常表示未正确初始化 CUDA,需要在代码中添加以下代码来初始化 CUDA:
```
#include <cuda_runtime.h>
#include <cublas_v2.h>
int main() {
// 初始化 CUDA
cudaError_t cudaStatus = cudaSetDevice(0);
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaSetDevice failed! Do you have a CUDA-capable GPU installed?");
return 1;
}
// 初始化 cuBLAS
cublasHandle_t handle;
cublasStatus_t cublasStatus = cublasCreate(&handle);
if (cublasStatus != CUBLAS_STATUS_SUCCESS) {
fprintf(stderr, "cublasCreate failed!");
return 1;
}
// ...
}
```
如果你已经添加了上述代码但仍然出现此错误,请确保你的 CUDA 和 cuBLAS 版本与你的代码兼容。
阅读全文