好看的企业网站首页WordPress二级栏目代码
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
