0
点赞
收藏
分享

微信扫一扫

centos7.9安装cuda驱动

斗米 2023-07-13 阅读 61
CUDA运维

环境

centos7.9,cuda

步骤

在CentOS 7.9上安装NVIDIA CUDA的步骤如下:

  1. 确保您的系统上没有任何NVIDIA驱动程序。
nvcc --version
或
nvcc -V

如果无输出则是未装驱动,如果已有,可以通过运行以下命令来完成卸载操作:

sudo yum remove nvidia-*

但是,在某些情况下,可能需要卸载所有与NVIDIA相关的内容,可以尝试使用以下命令:

sudo yum remove -y NVIDIA*
  1. 更新系统软件包
sudo yum update -y
  1. 安装开发工具:
sudo yum install -y gcc gcc-c++ make kernel-devel kernel-headers 
flex bison autoconf automake git libtool zlib1g-devel openssl-devel 
ncurses-devel ncurses-devel-static python-devel python-pip
  1. 安装cuda Python包: 首先,请确保您已经安装了Python和pip。如果没有,您可以使用以下命令安装它们:
sudo yum install python
sudo yum install python-pip

接下来,使用以下命令安装cuda Python包:

sudo pip install cuda-repo-rhel7-10-0-local
sudo yum install cuda-repo-10-0-local
sudo yum install cuda

注意:您需要将10-0-local替换为对应的CUDA版本。 5. 配置cuda环境变量: 打开/etc/profile文件,并在文件末尾添加以下内容:

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin

保存并关闭文件。然后,运行以下命令使更改生效:

source /etc/profile
  1. 测试CUDA安装: 运行以下命令来检查是否安装了CUDA以及其版本:
nvcc --version
或
nvcc -V

如果安装成功,您将看到类似于以下内容的输出:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Jun_13_19:16:58_PDT_2023
Cuda compilation tools, release 12.2, V12.2.91
Build cuda_12.2.r12.2/compiler.32965470_0

image.png 恭喜!您已成功在CentOS 7.9上安装了NVIDIA CUDA。现在,您可以使用它来编译和运行针对CUDA加速的GPU程序了。

举报

相关推荐

0 条评论