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

鱼满舱

关注

阅读 61

2023-01-17


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

1.下载cuda10.1

​​CUDA Toolkit 10.1官方下载地址​​

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

2.下载cudnn7.6.4

​​cuDNN官方下载地址​​

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

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

3.安装cuda10.1

sudo bash cuda_10.1.105_4188.39_linux.run

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

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

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

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

4.添加软链接

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

5.配置环境变量

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

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

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

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

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


精彩评论(0)

0 0 举报