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

指令編譯, 運作結果如下: