MindSpore GPU版本安装指导

MindSpore GPU软件包依赖Nvidia GPU驱动，CUDA runtime以及CUDNN库

当前MindSpore-GPU适配了CUDA 10.1和CUDA 11.1两个版本，可以根据操作系统版本和MindSpore版本选择对应的CUDA  Toolkit版本进行安装。

接下来以CUDA11.1 + CUDNN8.0为例，介绍如何安装上述软件

1、Nvidia GPU驱动和CUDA runtime可以通过CUDA Toolkit安装：

1）下载CUDA Toolkit

下载路径：https://developer.nvidia.com/cuda-11.1.1-download-archive

如下所示，当选择Ubuntu18.04版本的runfile之后，Base installer中会给出下载的路径以及安装命令

2）安装CUDA Toolkit

sudo sh cuda_11.1.1_455.32.00_linux.run

3）检查安装结果

执行nvidia-smi命令，可以看到当前cuda驱动版本为455.32.00，CUDA版本是11.1，默认安装路径是/usr/local/cuda-11.1

2、安装CUDNN

CUDNN是Nvidia提供的能神经网络算子库

下载CUNN之前需要完成账户注册（这里不再赘述），接下来在https://developer.nvidia.com/rdp/cudnn-archive选择对应的CUDNN版本

下载解压后分别将include和lib64移动到/usr/local/cuda/include目录即完成安装：