5

我正在尝试 CUDA 编程的“hello world”程序：将两个向量相加。这是我尝试过的程序：

#include <cuda.h>
#include <stdio.h> 
#define  SIZE 10

__global__  void vecAdd(float* A, float* B, float* C) 
{ 
   int i = threadIdx.x; 
   C[i] = A[i] + B[i]; 
} 

int  main() 
{ 
     float A[SIZE], B[SIZE], C[SIZE]; 
     float *devPtrA, *devPtrB, *devPtrC; 
     size_t memsize= SIZE * sizeof(float); 

     for (int i=0; i< SIZE; i++) {
        A[i] = i;
        B[i] = i;
     }

     cudaMalloc(&devPtrA, memsize); 
     cudaMalloc(&devPtrB, memsize); 
     cudaMalloc(&devPtrC, memsize); 
     cudaMemcpy(devPtrA, A, memsize,  cudaMemcpyHostToDevice); 
     cudaMemcpy(devPtrB, B, memsize,  cudaMemcpyHostToDevice); 

     vecAdd<<<1, SIZE>>>(devPtrA,  devPtrB, devPtrC); 
     cudaMemcpy(C, devPtrC, memsize,  cudaMemcpyDeviceToHost); 

     for (int i=0; i<SIZE; i++) 
         printf("C[%d]: %f + %f => %f\n",i,A[i],B[i],C[i]); 

     cudaFree(devPtrA); 
     cudaFree(devPtrB); 
     cudaFree(devPtrC); 
}

编译：

nvcc cuda.cu

输出是这样的：

C[0]: 0.000000 + 0.000000 => 0.000000
C[1]: 1.000000 + 1.000000 => 0.000000
C[2]: 2.000000 + 2.000000 => 0.000000
C[3]: 3.000000 + 3.000000 => 0.000000
C[4]: 4.000000 + 4.000000 => 0.000000
C[5]: 5.000000 + 5.000000 => 0.000000
C[6]: 6.000000 + 6.000000 => 0.000000
C[7]: 7.000000 + 7.000000 => 0.000000
C[8]: 8.000000 + 8.000000 => 366987238703104.000000
C[9]: 9.000000 + 9.000000 => 0.000000

每次运行它时，我都会得到 C[8] 的不同答案，但所有其他元素的结果始终为 0.000000。

Ubuntu 11.04 系统是一个 64 位 Xeon 服务器，具有 4 个内核，运行最新的 NVIDIA 驱动程序（2012 年 10 月 4 日下载）。该卡是具有 96 个内核和 1GB 内存的 EVGA GeForce GT 430。

我应该怎么做才能弄清楚发生了什么？

4

2 回答 2

7

似乎您的驱动程序未初始化，但不检查 cuda 返回代码始终是一种不好的做法，您应该避免这种做法。这是可用于 cuda 调用的简单函数 + 宏（引自 Cuda by Example）：

static void HandleError( cudaError_t err,
                         const char *file,
                         int line ) {
    if (err != cudaSuccess) {
        printf( "%s in %s at line %d\n", cudaGetErrorString( err ),
                file, line );
        exit( EXIT_FAILURE );
    }
}
#define HANDLE_ERROR( err ) (HandleError( err, __FILE__, __LINE__ ))

现在开始调用你的函数，如：

HANDLE_ERROR(cudaMemcpy(...));
于 2012-10-09T06:28:32.420 回答
1

最可能的原因：未加载 NVIDIA 驱动程序。在无头 Linux 系统上，X Windows 没有运行，因此驱动程序不会在启动时加载。

以 root 身份运行nvidia-smi -a以加载它们并以报告的形式获得确认。

尽管现在已加载驱动程序，但每次运行 CUDA 程序时仍需要对其进行初始化。将驱动程序置于持久模式，nvidia-smi -pm 1以便它们始终保持初始化状态。将此添加到启动脚本（例如 rc.local）中，以便在每次启动时发生。

于 2012-10-09T04:50:36.067 回答