在Ubuntu环境下安装CUDA,可以按照以下步骤进行:
-
下载CUDA Toolkit:从NVIDIA官方网站下载适用于Ubuntu的CUDA Toolkit安装包。
-
关闭图形界面:在安装CUDA前,最好在命令行界面下执行操作,可以通过按下
Ctrl + Alt + F1
切换到命令行终端。 -
安装依赖:在命令行终端中运行以下命令安装CUDA所需的依赖库:
sudo apt update sudo apt install build-essential
-
运行安装脚本:在命令行终端中进入下载的CUDA Toolkit安装包所在目录,并执行以下命令运行安装脚本:
chmod +x cuda_*.run sudo ./cuda_*.run
注意,
*.run
需要替换为下载的CUDA Toolkit安装包的版本号。 -
安装过程中会提示同意许可协议、选择安装位置和是否安装NVIDIA显卡驱动等选项,根据实际需求进行选择。
-
设置环境变量:安装完成后,需要将CUDA的路径添加到环境变量中。在
~/.bashrc
文件中添加以下内容:export PATH=/usr/local/cuda-X.X/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-X.X/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
注意,
X.X
需要替换为安装的CUDA版本号。 -
更新环境变量:执行以下命令使环境变量生效:
source ~/.bashrc
-
验证安装:运行以下命令查看CUDA版本:
nvcc --version
运行以下命令查看NVIDIA驱动版本:
nvidia-smi
安装完成后,CUDA就成功安装在Ubuntu系统中了,你可以开始使用CUDA来进行GPU加速的计算