ubuntu 18.04 安装PaddlePaddle GPU版本
ubuntu 18.04 安装PaddlePaddle GPU版本
- 一.安装cuda以及cudnn
-
- 1.1安装cuda
- 1.2安装cudnn
- 二.安装PaddlePaddle
-
- 2.1安装PaddlePaddle
本文章写于2020年9月19日16:57分
源于百度飞桨的目标检测课程作业（免费哦~），课程链接:https://aistudio.baidu.com/aistudio/education/group/info/1617
一.安装cuda以及cudnn
本例中使用的cuda和cudnn安装包百度网盘链接: https://pan.baidu.com/s/1F5qWXaBwpZfg8Flecc_eng 密码: a367
1.1安装cuda
笔者使用的是CUDA Toolkit 10.1(Feb 2019)为例进行安装(附cuda下载地址：https://developer.nvidia.com/cuda-toolkit-archive）。
由于笔者的采用的是ubuntu18.04操作系统，所以按照下图配置进行下载。
在cuda_10.1.105_418.39_linux.run所在文件夹处点击鼠标右键选择Open in Termainal打开终端，输入：
sudo sh cuda_10.1.105_418.39_linux.run
系统会提示您输入root密码，输入即可（输入的密码不可见哦，只管输入就好啦~）
接下来终端会弹出上图界面，我们输入accept后回车确认，会出现下面的界面，按上下方向键选择安装项目，对于已将装好nvidia显卡驱动的小伙伴可以不勾选Driver选项（选中按空格键可以取消勾选哦），其余的都是需要进行安装的。
接下来选中Install，按回车键开始安装，让我们静静等待~~~~~~~~~
安装完成后在终端输入 cat /usr/local/cuda/version.txt可以查看安装的cuda版本。
cat /usr/local/cuda/version.txt
至此我们的cuda就安装成功啦。
1.2安装cudnn
我们要登录nvidia的网站来下载cudnn(附网站链接：https://developer.nvidia.com/rdp/cudnn-archive)
小伙伴们要注意对于cuda和cudnn是有对应关系的，由于我们刚才下载的是cuda10.1，所以我们只能选择for cuda10.1版本的cudnn。
笔者选的是cuDNN v7.6.5(November 5th 2019),for CUDA 10.1（GPU 版本的 PaddlePaddle 需要使用 cuDNN 7.6+ 版本哦~），再选择cuDNN Library for Linux，下载即可。
找到我们下载的文件，名字叫做cudnn-10.1-linux-x64-v7.6.5.32.tgz，将其解压（选中下载文件，右键选择Extract Here,当然命令行解压也可以），双击进入解压生成的文件夹，点击鼠标右键选择Open in Termainal打开终端，执行如下命令：
# 复制cudnn头文件
sudo cp cuda/include/* /usr/local/cuda-10.1/include/
# 复制cudnn的库
sudo cp cuda/lib64/* /usr/local/cuda-10.1/lib64/
# 添加可执行权限
sudo chmod +x /usr/local/cuda-10.1/include/cudnn.h
sudo chmod +x /usr/local/cuda-10.1/lib64/libcudnn*
以上就安装好了cudnn，下面我们来测试下吧，输入下面的命令：
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
当出现了下图的内容，说明安装成功啦~~~~
恭喜你，你已经成功的安装了cuda和cudnn~~
二.安装PaddlePaddle
2.1安装PaddlePaddle
首先我们需要利用conda建立一个环境（需要先安装Anaconda），下面命令中的paddlepaddle是我给这个环境取得名字，小伙伴们可以按照自己的喜好取名字啊，python=后的的数字指定的Python版本，根据paddle官网的要求，Python 3 版本可以是 3.5.1+/3.6/3.7，笔者在这里就选择了3.6的版本。
conda create -n paddlepaddle python=3.6
环境建立完成后，我们在终端输入下面的代码激活环境，其中的padddlepaddle是我在建立环境时取得名字，记得换成你给环境取的名字哦。
conda activate paddlepaddle
命令行的开头变成你的环境的名字，就说明环境激活成功啦，下面我们开始安装paddle，输入下面命令。
python3 -m pip install paddlepaddle-gpu==1.8.4.post107 -i https://mirror.baidu.com/pypi/simple
这是只需要静静等待就好啦~
安装完成后我们来测试下是否安装成功吧，依次输入如下命令：
python
import paddle.fluid
paddle.fluid.install_check.run_check()
如果出现下面的内容（Your Paddle Fluid is installed successfully! ）恭喜你，你已经成功的安装了paddle~~~~~~~~~~
