1.打开网页https://developer.nvidia.com/cuda-downloads选择目标平台
在ubutnu 中使用wget命令下载:
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
即可完成下载:
在完成下载之后安装cuda10.2 。
输入命令安装cuda10.2
sudo sh cuda_10.2.89_440.33.01_linux.run
之后等待片刻之后会弹出安装界面:
按照要求输入accept,之后进入选择安装组件,这里不安装驱动程序,所以驱动程序前面的x用enter键取消掉:
使用键盘上的上下键移动光标至install,按下enter键即可等待完成安装:
在一下弹出的窗口中按照以下光标位置来选择:
之后等待片刻即可完成安装。
安装好之后会有以下提示:
安装好cuda之后开始设置环境变量。由于只是SSH连接的命令行界面。所以直接考虑使用vim或者winscp协助打开环境变量文件:
sudo gedit /etc/profile
原始的命令如上,可以使用winscp协助打开这个文件。打开之后在文件的最后面加入两行:
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
如下图所示:
完成之后输入命令是其临时生效:
source /etc/profile
之后输入命令查看cuda及其相关组件是否安装完成
cat /proc/driver/nvidia/version
nvcc -V
两个命令输入后其结果如下:
至此,CUDA10.2安装完成。
完成cuda10.2之后安装cudnn7.6.5,如下:打开cudnn下载网址:
https://developer.nvidia.com/cudnn
之后点击下载cuDNN,这里需要你注册登录。注册登录后选择适合cuda10.2的cudnn7.6..5进行下载:
在第三行里面的cudnn library for linux 上右击复制链接,使用ubuntu上的wget命令下载cudnn
wget https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.2_20191118/cudnn-10.2-linux-x64-v7.6.5.32.tgz?RA_4nMggwWRSMbmoscOgmb1AXT0WnGpn3nqwTZSwEOLSDBoU2ABEVUO1ieZkm1Ml-C2kRfLae_Zoho3qPBCkU_yKyVLuAjFXflXTloUN2e8XLPqC0A7AhBj4G1GQ9EvbH8gSnk6xYiy2LJRqPIfSsNnpbv_vLoPUtmPAD2rFhIwvZdivT5SqnfwFi6OVpC5NZafBknkWGfL4ynFvt1L4seAhJMqSoEL4Dg
下载完成之后
对原文件进行重命名之后再进行解压:
使用命令对压缩包进行解压
tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
解压的示意图如下:
之后执行一下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
如下图:
之后输入命令查看cudnn是否安装成功:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如图所示,显示的版本是cudnn7.6.5
即可完成CUDA+CUDNN的安装。
网友评论