Tensorflow学习笔记ubuntu

Ubuntu18.04 -cuda + cudnn安装

2020-07-23  本文已影响0人  h小吴y

1.在google cloud虚拟机中安装cuda10

在虚拟机中运行命令:

vim installNvidia.sh

在vim编辑器中按a键进入编辑模式,将下面一段代码复制到文件中。

选择自己对应的cuda版本:

https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

#!/bin/bash

echo "Checking for CUDA and installing."

# Check for CUDA and try to install.

if ! dpkg-query -W cuda; then

  # The 18.04 installer works with 18.10.

  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

  sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

  sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

  sudo apt-get update

  sudo apt-get install cuda-10.0

fi

echo 'export CUDA_HOME=/usr/local/cuda-10.0' >> ~/.bashrc

echo 'export PATH=$PATH:$CUDA_HOME/bin' >> ~/.bashrc

echo 'export LD_LIBRARY_PATH=$CUDA_HOME/lib64' >> ~/.bashrc

source ~/.bashrc

Esc键进入vim编辑器的命令模式,输入:wq并运行,表示保存并退出。

在虚拟机中运行命令:

bash installNvidia.sh

运行少量时间后,如下图所示,询问是否继续,回答y

经过大概10分钟,可以安装成功。

查看Nvidia显卡的使用情况,命令:nvidia-smi

查看CUDA版本号,命令:

cat /usr/local/cuda/version.txt

如果显示下图,则表示安装CUDA10.0成功

其它安装方法:

安装驱动

https://blog.csdn.net/sss_369/article/details/94591280

https://blog.csdn.net/m0_37638193/article/details/79492174

安装多版本cuda

解决cudnn版本不对应问题

2.安装cudnn

下载地址:https://blog.csdn.net/xiangxianghehe/article/details/79177833

linux系统中下载命令为:wget {},大括号替换为下载链接。

wget http://file.ppwwyyxx.com/nvidia/cudnn-10.0-linux-x64-v7.6.4.38.tgz 

linux系统中解压tgz压缩文件命令为:tar -zxvf {},大括号替换为tgz压缩文件名。

sudo tar -zxvf cudnn-10.0-linux-x64-v7.6.4.38.tgz

解压文件后,路径下有一个cuda文件夹,运行ls命令,如下图所示:

将cuda文件夹中的文件复制到linux的cuda程序中,两行命令如下:

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include/

sudo cp cuda/lib64/* /usr/local/cuda-10.0/lib64/

查看cudnn版本

在终端输入

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

到这里为止,就已经安装cuDNN成功。

参考链接:https://www.jianshu.com/p/893d622d1b5a

上一篇下一篇

猜你喜欢

热点阅读