在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加速的计算。请注意,安装CUDA可能会影响系统的稳定性,建议在安装前备份重要数据,并确保对操作有足够的了解。如果不确定如何操作,建议寻求专业的帮助或咨询。