ubuntu18.04安装cuda

在Ubuntu 18.04上安装CUDA需要完成以下几个步骤：

第一步：安装NVIDIA显卡驱动程序。

在终端中输入以下命令来添加NVIDIA显卡驱动程序的PPA：

sudo add-apt-repository ppa:graphics-drivers/ppa

然后更新包列表并安装最新的NVIDIA驱动程序：

sudo apt update
sudo apt install nvidia-driver-418

安装完成后，重启系统以使驱动程序生效。

第二步：下载并安装CUDA_TOOLKIT。

在终端中输入以下命令来下载CUDA_Toolkit。使用ctrl+alt+t打开终端。需要了解CUDA_TOOLKIT的版本和GPU的计算能力，可以在以下网址查看：https://developer.nvidia.com/cuda-gpus。

cd ~/Downloads
wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux

然后下载并安装CUDA_Toolkit：

sudo chmod +x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run –override

安装完成后，可能需要设置一些环境变量：

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

如果您使用的是其他版本的CUDA_Toolkit，请相应调整上面的命令。

第三步：编写并编译CUDA程序。

安装CUDA_Toolkit后，可以使用nvcc命令来编写和编译CUDA程序。您可以使用以下命令来测试CUDA是否正确安装：

echo "__device__ int gcd(int a, int b) { return b == 0 ? a : gcd(b, a % b); } __global__ void saxpy(int n, float a, float *x, float *y) { int i = blockIdx.x * blockDim.x + threadIdx.x; y[i] = a * x[i] + y[i]; } int main() { const int N = 1 << 20; float *x, *y, *d_x, *d_y; x = (float*)malloc(N*sizeof(float)); y = (float*)malloc(N*sizeof(float)); cudaMalloc(&d_x, N*sizeof(float)); cudaMalloc(&d_y, N*sizeof(float)); for (int i = 0; i < N; i++) { x[i] = 1.0f; y[i] = 2.0f; } cudaMemcpy(d_x, x, N*sizeof(float), cudaMemcpyHostToDevice); cudaMemcpy(d_y, y, N*sizeof(float), cudaMemcpyHostToDevice); saxpy<<<(N+255)/256, 256>>>(N, 2.0f, d_x, d_y); cudaMemcpy(y, d_y, N*sizeof(float), cudaMemcpyDeviceToHost); cudaFree(d_x); cudaFree(d_y); free(x); free(y); return gcd((int)y[0], N); }" > test.cu

然后使用以下命令来编译和运行测试CUDA程序：

nvcc test.cu -o test
./test

如果运行结果为1，则表示CUDA已成功安装并配置。