Ubuntu 20.04安装CUDA 11

一、安装Nvidia显卡驱动

• 由于Ubuntu软件和更新里的驱动不支持Cuda11，最高支持10.2，所以到Nvidia官网安装最新驱动，我的最新版本号是450.57。

• 输入显卡信息，搜索
  

• 搜索结果，下载
  

• 下载完成后将文件移动到主目录下（别问为什么，后面就知道了），我放在主目录下。
  

• 然后，请一次性看完并记住下面几步，或者用手机打开这条教程。

  • 打开终端（Ctrl+Alt+T）输入：
    sudo telinit 3
  • 然后你的电脑就关闭了图形界面，进入了tty界面
  • 输入用户名和密码。（注意：输入数字时不要用小键盘，使用字母上面那一排）
  • 输入以下命令：

    ls
    # ls查看文件名，方便输入。
    sudo sh 文件名
    

  • 然后就进入了显卡安装界面，疯狂回车就好了
  • 安装完之后，输入nvidia-smi检查驱动是否安装成功
  • 然后输入命令startx回到图形界面

• 到这里Nidia驱动就安装好了，将文件放到主目录下是因为tty界面不能输入和显示中文，如果文件在/下载目录下，就访问不到了。

二、安装Cuda11

• 由于我在安装Cuda的时候，下载完成后出现了s段已转储错误，所以建议先执行以下步骤，避免此错误：
  • gedit .bashrc
  • 在文件最后添加：

    ulimit -c unlimited
    ulimit -s 819200
    

  • 保存文件退出，重启另一个新的终端，输入以下命令激活：

    source .bashrc
    

• 获取cuda安装命令：https://developer.nvidia.com/cuda-downloads
  
• 依次输入这两条命令

  wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
  sudo sh cuda_11.0.2_450.51.05_linux.run
  

• 第一条命令结束之后如果出现s段错误，没关系，不用理他。
• 第二条命令里除了有一步输入accpet，以及在很多X选择的时候，把第一个Nvidia显卡去掉（按回车去掉），其他都是默认，按回车就行。
• 安装完毕之后添加环境变量

  gedit ~/.bashrc
  # 文本最后添加以下内容：
  export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
  export PATH=$PATH:/usr/local/cuda/bin
  export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
  # 保存退出，打开新终端激活
  source ~/.bashrc
  

• 测试Cuda

  cd /usr/local/cuda/samples/1_Utilities/deviceQuery
  make
  ./deviceQuery
  

• Result = PASS则安装成功
  

三、安装cuDNN

• 点击链接：Nvidia cuDNN
• 需要登陆Nvidia账号，选择Download cuDNN，选择for Cuda11的版本下载。我下载的第一个。
  
• 下载完成后解压，进入解压后的目录，执行

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

• 安装完成