Ubuntu18.04下安装cuda和cudnn
时间: 2023-08-19 16:06:22 浏览: 197
ubuntu安装显卡驱动+cuda+cudnn教程(18.04+cuda10+cudnn9.05为例)
5星 · 资源好评率100%
您好!在Ubuntu 18.04上安装CUDA和cuDNN需要以下步骤:
1. 安装NVIDIA显卡驱动:
首先,您需要安装适用于您的显卡型号的NVIDIA驱动程序。您可以通过以下命令来安装:
```shell
sudo ubuntu-drivers autoinstall
```
这将自动检测并安装适合您显卡型号的驱动程序。
2. 下载并安装CUDA:
访问NVIDIA的官方网站,并下载适用于Ubuntu 18.04的CUDA安装包。您可以在https://developer.nvidia.com/cuda-downloads 上找到适合您系统的CUDA版本。选择合适的版本并下载安装包。
安装CUDA之前,您需要先关闭图形界面。按Ctrl+Alt+F1切换到TTY模式,然后在终端中输入以下命令以停止图形界面服务:
```shell
sudo service lightdm stop
```
运行安装包并按照提示进行安装:
```shell
sudo sh cuda_*.run
```
安装完成后,重新启动系统。
3. 配置环境变量:
将CUDA的bin和lib目录添加到系统的环境变量中。编辑`~/.bashrc`文件,添加以下行:
```shell
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
保存文件并执行以下命令使配置生效:
```shell
source ~/.bashrc
```
4. 下载并安装cuDNN:
访问NVIDIA的官方网站,并下载适用于Ubuntu 18.04的cuDNN安装包。您需要注册为NVIDIA开发者才能下载cuDNN。
下载完成后,解压缩安装包并进入解压缩后的目录:
```shell
tar -xzvf cudnn-*.tgz
cd cuda
```
将cuDNN的库文件复制到CUDA的安装目录中:
```shell
sudo cp -P include/cudnn.h /usr/local/cuda/include
sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
```
5. 完成!
现在您已经成功安装了CUDA和cuDNN。您可以编译并运行依赖于CUDA的程序。
请注意,上述步骤仅适用于Ubuntu 18.04和支持NVIDIA GPU的系统。确保您的显卡型号与CUDA和cuDNN的要求相匹配。
阅读全文