ubuntu 安装CUDA cudnn
时间: 2024-06-11 21:01:46 浏览: 127
首先,Ubuntu是一种开源操作系统,CUDA和cuDNN是用于深度学习的重要库。下面是在Ubuntu上安装CUDA和cuDNN的步骤:
1. 下载CUDA
在NVIDIA官网上下载适合您的CUDA版本。可以选择.deb或.run文件。
2. 安装CUDA
对于.deb文件,使用以下命令安装:
```
$ sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda
```
对于.run文件,使用以下命令安装:
```
$ chmod +x cuda_9.0.176_384.81_linux.run
$ sudo ./cuda_9.0.176_384.81_linux.run
```
3. 设置环境变量
添加以下行到文件$HOME/.bashrc中:
```
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
4. 下载cuDNN
在NVIDIA官网上下载与CUDA版本对应的cuDNN库。需要注册并登录。
5. 安装cuDNN
解压下载的cuDNN库,并将文件复制到CUDA安装目录下的相应文件夹中:
```
$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
$ sudo cp -P cuda/include/cudnn.h /usr/local/cuda-9.0/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
$ sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*
```
阅读全文