关于NVIDIA显卡的驱动安装, 可以参考旧文, 装好驱动之后, 只需要一条指令即可完成cuda环境的搭建.
sudo apt install nvidia-cuda-toolkit
大小在2G左右, 和官网下载的安装包大小相当. 然后用nvcc -V进行查看, 发现没有安装最新的CUDA10.0, 安装的是CUDA9.1. 操纵系统是Ubuntu18.04LTS.
CUDA版本
然后使用如下代码进行测试:
#include <stdio.h>
__global__ void helloFromGPU ()
{
printf("Hello, world! from GPU!\n");
}
int main()
{
printf("Hello, world! from CPU!\n");
helloFromGPU <<<2, 10>>>();
cudaDeviceReset();
return 0;
}
使用
nvcc HWGPU.cu -o HWGPU.o
指令编译, 运行结果如下:
运行结果