Ubuntu18.04LTS快速搭建CUDA环境

关于NVIDIA显卡的驱动安装, 可以参考旧文, 装好驱动之后, 只需要一条指令即可完成cuda环境的搭建.

sudo apt install nvidia-cuda-toolkit
复制代码

大小在2G左右, 和官网下载的安装包大小相当. 然后用nvcc -V进行查看, 发现没有安装最新的CUDA10.0, 安装的是CUDA9.1. 操纵系统是Ubuntu18.04LTS.

然后使用如下代码进行测试:

#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指令编译, 运行结果如下: