准备
如果你是第一次使用CUDA,在Linux下可以使用下面的命令来检查CUDA编译器是否安装正确:
$ which nvcc
一般,该指令输出为:
/usr/local/cuda/bin/nvcc
另外,你可能还需要检查下你机器上的GPU型号,可以使用给下面的命令查询:
$ ls -l /dev/nv*
可能的输出为:
crw-rw-rw- 1 root root 195, 0 Jul 3 13:44 /dev/nvidia0
crw-rw-rw- 1 root root 195, 1 Jul 3 13:44 /dev/nvidia1
crw-rw-rw- 1 root root 195, 255 Jul 3 13:44 /dev/nvidiactl
crw-rw---- 1 root root 10, 144 Jul 3 13:39 /dev/nvram
以上输出显示这里有两个GPU显卡安装在机器上。
写一段CUDA程序的基本过程为:
- 创建源文件,以”.cu”为后缀。
- 用nvcc编译程序。
- 命令行运行。
然后是我们所谓的kernel function(即CUDA代码)
__global__ void helloFromGpu(void){
printf(“hello world form GPU!\\n”);
}
限定符__global__告诉编译器这个function将由CPU调用在GPU上执行,其调用形式为:
helloFromGPU<<