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
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成功。