目录
有的时候发现最新的包无法安装
查看下内核版本是否支持
步骤1:下载CUDA 12.6.1的安装包
你可以尝试使用以下命令来下载:
wget https://developer.download.nvidia.com/compute/cuda/12.6.1/local_installers/cuda_12.6.1_560.35.03_linux.run
如果URL有变更,访问NVIDIA官网获取最新的下载链接。
步骤2:赋予安装包执行权限
下载完成后,赋予安装包执行权限:
chmod +x cuda_12.6.1_560.35.03_linux.run
步骤3:运行安装包
运行安装包进行安装:
sudo ./cuda_12.6.1_560.35.03_linux.run
在安装过程中,你将看到多个选项。通常,你可以接受默认选项,往下选择install,然后enter回车,要等一会。
更新的时候需要连驱动一同更新
步骤4:配置环境变量
打开或创建 ~/.bashrc
文件,并添加以下行:
export PATH=/usr/local/cuda-12.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并关闭文件,然后更新当前 shell 的环境变量:
source ~/.bashrc
步骤5:验证安装
你可以通过运行以下命令来验证 CUDA 是否安装正确:
sh复制代码
nvcc -V
你应该能看到 CUDA 12.6.1 的编译器版本信息。
步骤6:检查驱动兼容性
通常情况下,CUDA安装包会包含适合的NVIDIA驱动程序。如果你的系统已经有较新的驱动,你可能会跳过驱动安装部分。你可以使用以下命令检查当前驱动版本:
sh复制代码
nvidia-smi
确保你的驱动版本与CUDA 12.6.1兼容。
步骤7:测试兼容性
编译并运行一个简单的CUDA程序来测试安装是否成功。例如,你可以创建一个简单的CUDA程序文件test.cu
,内容如下:
cpp复制代码
#include <stdio.h>
__global__ void hello_from_gpu() {
printf("Hello World from GPU!\n");
}
int main() {
hello_from_gpu<<<1, 1>>>();
cudaDeviceSynchronize();
return 0;
}
编译并运行:
sh复制代码
nvcc test.cu -o test
./test
如果一切正常,你应该能看到程序输出 "Hello World from GPU!"。