我正在尝试CUDA编程的“hello world”程序:将两个向量加在一起。以下是我已经尝试过的节目:编译时无法获得简单的CUDA程序
#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系统是一款运行最新NVIDIA驱动程序的4核64位Xeon服务器(2012年10月4日下载)。该卡是EVGA GeForce GT 430,具有96个内核和1GB的RAM。
我该怎么做才能弄清楚发生了什么事?
你从[这里]安装CUDA软件开发包(http://developer.nvidia.com/cuda/cuda-downloads)? (不是工具箱) – none
我的猜测是CUDA无法初始化。我建议为每个CUDA API调用添加错误检查。 – njuffa
@gokcehan我从该网站下载了驱动程序,工具包和SDK。不过,我不确定如何处理SDK。它似乎主要包含文档和示例代码。 –