ubuntu安装cudnn和cuda
时间: 2024-12-26 11:20:35 浏览: 15
### Ubuntu系统下安装CUDA和cuDNN
#### 安装前准备
为了确保顺利安装CUDA和cuDNN,在开始之前需确认系统的驱动程序已更新至支持CUDA的版本。可以通过NVIDIA官方网站查询当前显卡所适配的最新驱动并完成安装。
#### 下载CUDA工具包
访问[NVIDIA官方CUDA下载页面](https://developer.nvidia.com/cuda-downloads),选择适合的操作系统版本(此处为Ubuntu),以及相应的架构和其他选项来获取适用于环境的具体链接或指令集[^1]。
#### 安装CUDA
对于大多数情况来说,建议采用.run文件形式来进行CUDA Toolkit的手动部署。这允许更灵活地控制安装位置等参数。执行如下命令可以启动安装过程:
```bash
sudo sh cuda_<version>_linux.run
```
按照提示操作即可完成基本设置。完成后记得重启计算机使更改生效,并通过`nvcc --version`验证安装成功与否。
#### 配置环境变量
编辑~/.bashrc或其他shell配置文件,加入以下路径以便于后续调用编译器及其他组件:
```bash
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc
```
#### 获取cuDNN库
前往[NVIDIA开发者登录页](https://developers.nvidia.com/rdp/form/cudnn-download-survey)注册账号后可获得不同版本的cuDNN资源。注意要挑选与先前安装好的CUDA相匹配的那个版本号下的`.tgz`压缩包进行下载[^2]。
#### 解压并复制文件
解压下载下来的cuDNN压缩包并将其中的内容移动到CUDA目录内:
```bash
tar -xzvf cudnn-<version>-linux-x64-v8.0.tar.xz
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
```
再次刷新环境变量使之立即可用:
```bash
source ~/.bashrc
```
此时已经完成了整个流程中的主要部分,可通过运行一些简单的测试样例检验是否一切正常工作。
阅读全文