ubuntu18.04安装cuda10.2cudnn
时间: 2023-04-30 19:00:27 浏览: 184
1. 首先下载cuda10.2和cudnn的安装包,可以从官网下载或者使用镜像站下载。
2. 安装cuda10.2,可以使用命令行或者图形界面安装。命令行安装可以使用以下命令:
```
sudo sh cuda_10.2.89_440.33.01_linux.run
```
安装过程中需要选择安装路径、是否安装驱动等选项,根据自己的需求选择即可。
3. 安装cudnn,将下载的cudnn安装包解压缩后,将其中的文件复制到cuda的安装目录中。假设cuda的安装目录为/usr/local/cuda-10.2,则可以使用以下命令进行复制:
```
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h /usr/local/cuda-10.2/lib64/libcudnn*
```
4. 配置环境变量,将cuda和cudnn的路径添加到环境变量中。可以在~/.bashrc文件中添加以下内容:
```
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
```
5. 最后,重新加载~/.bashrc文件,使环境变量生效:
```
source ~/.bashrc
```
至此,cuda10.2和cudnn的安装就完成了。可以使用nvcc -V命令检查cuda的版本,使用python -c "import tensorflow as tf; print(tf.__version__)"命令检查cudnn是否安装成功。
阅读全文