当安装好显卡驱动后怎么样知道驱动程序安装好了,这里以T400 + OpenEuler 正常情况下,我们只要看一下nvidia-smi 状态就可以确定他已经正常了
如图:
这里就已经确定是可以正常使用了,这里只是没有运行对应的程序,那接来下我们就写一个测试程序来测试一下:以下代码通过AI给出然后做了一些小改
这里做两个文件:
首先,让我们创建一个C文件,命名为`gpu_matrix_multiply.cu`:
#include <stdio.h>
#include <stdlib.h>
#include <cuda_runtime.h>
#define N 1024 // Matrix size (N x N)
#define BLOCK_SIZE 32
__global__ void matrixMultiply(float *A, float *B, float *C) {
int row = blockIdx.y * blockDim.y + threadIdx.y;
int col = blockIdx.x * blockDim.x + threadIdx.x;
float sum = 0.0f;
if (row < N && col < N) {
for (int i = 0; i < N; i++) {
sum += A[row * N + i] * B[i * N + col];
}
C[row * N + col] = sum;
}
}
void initMatrix(float *matrix) {
for (int i = 0; i < N * N; i++) {


5867

被折叠的 条评论
为什么被折叠?



