-
安装CUDA工具包:
- 下载CUDA工具包:
- 访问NVIDIA官方网站的CUDA下载页面,选择适合系统的版本并下载安装包。
- 安装CUDA工具包:
- 打开终端并导航到下载目录,运行以下命令:
sudo dpkg -i cuda-repo-<distro>_<version>_amd64.deb sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/<distro>/x86_64/7fa2af80.pub sudo apt-get update sudo apt-get install cuda
<distro>
和<version>
需要替换为具体的发行版和版本号。 - 打开终端并导航到下载目录,运行以下命令:
- 下载CUDA工具包:
-
配置环境变量:
- 安装完成后,需要将CUDA工具包的路径添加到环境变量中。编辑
~/.bashrc
文件,添加以下内容:
其中export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
<version>
是CUDA的版本号。 - 使更改生效:
source ~/.bashrc
- 安装完成后,需要将CUDA工具包的路径添加到环境变量中。编辑
-
验证CUDA安装:
- 使用以下命令验证CUDA是否正确安装:
nvcc -V
- 这将显示CUDA编译器的版本信息。
- 运行示例程序验证CUDA功能:
这将显示有关GPU设备的信息,验证CUDA是否正常工作。cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make ./deviceQuery
- 使用以下命令验证CUDA是否正确安装:
-
使用CUDA进行开发:
- 安装CUDA工具包后,可以使用CUDA进行GPU编程。常见的开发环境包括NVIDIA Nsight、Visual Studio Code等。
- 编写CUDA程序时,需要包含CUDA头文件并链接CUDA库。例如:
#include <cuda_runtime.h> #include <device_launch_parameters.h>
- 使用
nvcc
编译CUDA程序:nvcc -o my_cuda_program my_cuda_program.cu
-
更新和卸载CUDA:
- 更新CUDA工具包:
sudo apt-get update sudo apt-get upgrade
- 卸载CUDA工具包:
sudo apt-get purge cuda sudo apt-get autoremove
- 更新CUDA工具包: