Linux系统上安装NVIDIA CUDA的步骤如下:
CUDA(Compute Unified Device Architecture)是NVIDIA推出的一个并行计算平台和编程模型,它允许开发者使用NVIDIA的GPU来进行通用计算。
以下是在Ubuntu系统上安装CUDA的步骤:
sudo apt update
sudo apt upgrade
首先,添加NVIDIA PPA:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
然后,安装推荐的驱动版本:
sudo apt install nvidia-driver-<version>
重启系统使驱动生效:
sudo reboot
下载CUDA Toolkit的安装包(可以从NVIDIA官网获取最新版本的安装脚本):
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-1_amd64.deb
sudo apt-key add /var/cuda-repo-11-4-local/7fa2af80.pub
sudo apt update
sudo apt install cuda
编辑~/.bashrc
文件,添加以下行:
export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
然后,使环境变量生效:
source ~/.bashrc
~/.bashrc
文件中的路径设置,并确保执行了source ~/.bashrc
命令。以下是一个简单的CUDA C程序示例:
#include <stdio.h>
__global__ void hello_from_gpu() {
printf("Hello from GPU!\n");
}
int main() {
hello_from_gpu<<<1, 1>>>();
cudaDeviceSynchronize();
return 0;
}
编译并运行:
nvcc hello.cu -o hello
./hello
通过以上步骤,你应该能够在Linux系统上成功安装并运行CUDA。如果在安装过程中遇到问题,请参考NVIDIA官方文档或社区论坛寻求帮助。
没有搜到相关的文章