linux服务器安装cuda,cudnn
时间: 2025-01-17 11:03:55 浏览: 59
### 安装配置CUDA和cuDNN
#### 准备工作
为了在Linux服务器上成功安装并配置CUDA和cuDNN,在开始之前需确认已具备以下条件:
- 已获得管理员授权或具有sudo权限来执行必要的命令操作;如果确实无法获取超级用户权限,则可以尝试仅限于个人账户下的局部环境搭建,但这可能会影响某些功能正常使用。
- 确认目标机器已经预装了支持CUDA技术的NVIDIA显卡驱动程序。
#### 下载CUDA Toolkit
访问[NVIDIA官方网站](https://developer.nvidia.com/cuda-downloads),按照页面提示选择适合当前系统的版本下载。对于运行着CentOS 7.6.1810的操作系统而言,应当挑选对应此发行版以及硬件架构(x86_64)的CUDA 10.1 Runfile (Local)[^2]。
#### 执行安装过程
通过SSH连接到远程主机后,切换至保存有刚才所下载文件的位置,并赋予其可执行属性:
```bash
chmod +x cuda_<version>_linux.run
```
启动安装向导前建议先关闭X服务以免造成冲突(如果有图形界面的话),接着输入如下指令触发安装流程:
```bash
sudo ./cuda_<version>_linux.run
```
遵循屏幕上的指导完成整个设置过程,期间可能会询问是否要安装第三方组件比如NSight Visual Studio Code Extension等工具集,默认选项通常是合理的,可根据实际需求调整。
#### 设置环境变量
编辑`~/.bashrc`或其他shell初始化脚本文件,追加下面几行用来定义新的PATH与LD_LIBRARY_PATH路径指向新安装好的CUDA库目录下:
```bash
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
使更改立即生效可通过source加载该配置文件:
```bash
source ~/.bashrc
```
#### cuDNN集成
前往[NVIDIA cuDNN Archive](https://developer.nvidia.com/rdp/cudnn-archive),登录账号找到适用于上述CUDA版本号的cuDNN包进行下载解压。将其中的内容复制粘贴覆盖至先前指定的目标位置即完成了基本整合工作。
验证安装成果可以通过编写一段简单的测试代码调用GPU资源来进行检验,确保一切正常运作之后便大功告成啦!
阅读全文
相关推荐


















