Hi, I’m trying to use ceres-solver on my Jetson Orin. For this, I have a function to check weather my device has a GPU and if the GPU supports all the needed functions.
This code is stuck/forzen for 7 min when calling cublasCreate
My nvidia-smi returns:
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 540.3.0 Driver Version: N/A CUDA Version: 12.2 |
|-----------------------------------------+----------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+======================+======================|
| 0 Orin (nvgpu) N/A | N/A N/A | N/A |
| N/A N/A N/A N/A / N/A | Not Supported | N/A N/A |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
+---------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=======================================================================================|
| No running processes found |
+---------------------------------------------------------------------------------------+
The code is:
#include <iostream>
#include <cublas_v2.h>
int main() {
cublasHandle_t handle;
cublasStatus_t status = cublasCreate(&handle);
if (status == CUBLAS_STATUS_SUCCESS) {
std::cout << "cuBLAS is supported on this device." << std::endl;
cublasDestroy(handle);
} else {
std::cout << "cuBLAS is not supported on this device." << std::endl;
}
return 0;
}
Compiling: nvcc -o test_cublas test_cublas.cpp -lcublas
The output is: cuBLAS is supported on this device after 7 minutes.
Note that it takes also 7 min if I rerun this another time.