CudaToolKit安装(nvcc)
cudatoolkit
是 CUDA 开发工具包(CUDA Toolkit) 的核心部分,包含了一系列用于开发和运行 CUDA 应用程序的软件组件。nvcc
是 NVIDIA CUDA 编译器驱动,用于将 CUDA C/C++ 代码编译成可以在 NVIDIA GPU 上运行的可执行代码。nvcc
是 cudatoolkit
的一部分,需要安装完整的 CUDA Toolkit 才能使用。
安装完驱动后先查看当前是否能读取nvcc
nvcc -V
正常读取情况下,会显示如下内容
user@vrlab:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Tue_Oct_29_23:50:19_PDT_2024
Cuda compilation tools, release 12.6, V12.6.85
Build cuda_12.6.r12.6/compiler.35059454_0
如果没有显示则说明没有安装,如果显示版本过低,需要切换版本
- 安装cudatoolkit
- 修改版本
1、安装cudatoolkit
建议配置完镜像后再进行安装
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-6
下载完成后检查nvcc
版本
2、修改版本
如果版本不对(如安装了12.6,输出为10.1),则需要检查当前的nvcc映射路径
which nvcc
如果显示的路径不是 /usr/local/cuda-12.6/bin
,说明系统仍在使用旧版本的 CUDA。
更新环境变量
确保你的环境变量 PATH
和 LD_LIBRARY_PATH
指向新版本的 CUDA。
临时更新
在当前终端中执行以下命令:
export PATH=/usr/local/cuda-12.6/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH
然后检查 nvcc
的版本:
nvcc --version
如果版本显示正确,可以将上述命令添加到你的 ~/.bashrc
或 ~/.zshrc
文件中。
永久更新
编辑 ~/.bashrc
或 ~/.zshrc
文件:
vim ~/.bashrc
添加以下内容:
export PATH=/usr/local/cuda-12.6/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH
保存并退出,然后刷新环境变量:
source ~/.bashrc