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目录即完成安装: