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指令編譯, 運行結果以下: