linux centos安装 cuda
时间: 2024-08-16 16:07:10 浏览: 110
在 CentOS 系统上安装 CUDA 需要满足一些前提条件,并按照特定步骤操作。以下是基本的安装流程:
1. **检查硬件兼容性**:
CUDA 通常需要支持的 NVIDIA 显卡,先确保你的显卡型号在CUDA官网的兼容设备列表中。
2. **添加官方仓库**:
打开终端并运行命令:
```
sudo curl https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/repodata/repomd.xml | sudo tee /etc/yum.repos.d/cuda-rhel7.repo
```
3. **更新软件源**:
```
sudo yum makecache fast
```
4. **安装 CUDA**:
使用 `yum` 安装CUDA:
```
sudo yum install -y cuda-cudart-dev cuDNN-devel
```
这会安装CUDA的基本库和CUDDN(用于深度学习加速)。
5. **安装驱动**:
如果需要安装GPU驱动,你可以从NVIDIA官网下载合适的RPM包,然后通过`rpm`安装:
```
wget http://<download_link> -O nvidia-driver.rpm
sudo rpm -ivh nvidia-driver.rpm
```
6. **配置环境变量**:
完成安装后,编辑`~/.bashrc`文件添加以下行:
```sh
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc
```
7. **验证安装**:
使用 `nvcc --version` 检查CUDA版本是否已成功安装。
阅读全文