当前位置:   article > 正文

linux版本安装cuda_linux安装cuda

linux安装cuda

安装及配置过程

一、下载安装CUDA Toolkit

  • 1.查看操作系统版本及支持CUDA版本
  • 2.官网下载并安装对应版本CUDA
  • 3.配置环境变量
  • 4.测试CUDA安装是否成功
二、下载安装cuDNN
  • 1.官网下载对应版本cuDNN

一、下载安装CUDA Toolkit

1.查看操作系统版本及支持CUDA版本

1)查看系统版本

uname -a
  • 1

在这里插入图片描述
2)查看系统支持CUDA版本
图中标红处说明此系统支持CUDA最高版本为:11.0,即下载CUDA时版本要控制在11.0以下

nvidia-smi
  • 1

在这里插入图片描述


2.官网下载并安装对应版本CUDA

1)根据系统支持版本下载对应版本的CUDA Toolkit,作者此处选择CUDA10.2。官网链接
2)选择所需版本,通过对应命令进行下载安装(注意此处需要记住下载文件的目录,之后需要找到)
在这里插入图片描述按照顺序输入上图 下方的两条指令

在输入第二条命令之后,经过短暂的等待,会出现用户安装界面,
其中包括是否选择安装Nvidia显卡驱动,如果本地已有驱动,可选择不安装(将光标移至Driver,点击Enter即可),
之后移至Install处,点击Enter即可进行安装。
  • 1
  • 2
  • 3
3.配置环境变量

1)进入系统home文件夹,找到 .bashrc 文件,添加环境变量(其中涉及到版本问题,均与自己安装版本对应即可)

 export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
 export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  • 1
  • 2

2)激活配置文件

source ~/.bashrc
  • 1

3)如果本地安装了多个版本的CUDA,在切换版本时,可通过更改上述文件中CUDA版本即可,如将cuda-11.1改为cuda-10.1即可成功切换。

  1. 测试CUDA安装是否成功
    出现如下字样即说明安装成功
 nvcc -V
  • 1

在这里插入图片描述


二、下载安装cuDNN

1.官网下载对应版本cuDNN

1)对应上述安装的CUDA版本,选择合适的cuDNN版本,官网链接

  • 此处需要注册Nvidia账户
  • 要将下方的版本号修改成自己的版本

2)选中对应版本之后,下载对应Tar包(记住你的下载路径)

3)进入Tar包下载目录,对其进行解压(解压结果类似下图)

 tar -xvf cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive.tar.xz
  • 1

4)将cudnn安装目录下部分文件复制到cuda的安装目录下

sudo cp cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive/lib/libcudnn* /usr/local/cuda/lib64 
  • 1
  • 2

5)修改权限

sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
  • 1

参考链接:https://blog.csdn.net/qq_43665602/article/details/125752433

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小桥流水78/article/detail/972504
推荐阅读
相关标签
  

闽ICP备14008679号