Linux安装cuda10.1,cudnn7.6.4 (深度学习) MindSporeGPU计算框架

CUDA(ComputeUnified Device Architecture)，是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。当我们在进行深度学习任务时，我们有时需要使用GPU版本的框架，比如tensorflow-gpu或者mindspore-gpu版本就需要安装CUDA和CUDNN进行GPU加速支持

1.下载cuda10.1

CUDA Toolkit 10.1官方下载地址

2.下载cudnn7.6.4

cuDNN官方下载地址

注意要下载与cuda版本对应的cudnn工具

3.安装cuda10.1

sudo bash cuda_10.1.105_4188.39_linux.run

看到如下界面，在控制台输入accept进行下一步安装

然后再下述界面只需要勾选CUDA Toolkit 10.1即可，第一个为显卡驱动，如果当前驱动版本较低需要勾选安装

4.添加软链接

sudo cp ./include/* /usr/localcuda/include
sudo cp ./lib64/* /usr/local/cuda/lib64

5.配置环境变量

下图是已经安装好的cuda路径

vim ~/.bashrc
#cuda 配置cuda的安装路径，默认是在/usr/local
export CUDA_HOME=/usr/local/cuda-10.1
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
source ~/.bashrc

6.验证是否安装成功

nvcc -V

如果出现该提示，说明安装成功