ubuntu20.04+cuda+cudnn安装

前言:

CUDA及cuDNN的安装(linux)Linux下安装显卡驱动及CUDA程序

系统参数:

系统:ubuntu20.04.2

显卡: GeForce GTX 1050

需要安装的版本:CUDA 11.4

一、 安装显卡驱动

  (1) 检测你的NVIDIA图形卡和推荐的驱动程序的模型

ubuntu-drivers devices

  (2) 选择安装所有推荐的驱动

sudo ubuntu-drivers autoinstall

    (3) nvidia-smi      检查安装驱动是否成功

结果显示:

NVIDIA-SMI 470.74      Driver Version: 470.74      CUDA Version: 11.4        

二、 安装CUDA程序

   (1)下载cuda:  

进入NVIDIA下载安装包 CUDA Toolkit 11.7 Downloads | NVIDIA Developer

选择相应的版本,有相关的命令 在终端输入即可 (注意:输入自己的安装版本)

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

    (2) cuda安装:

sudo sh cuda_11.0.2_450.51.05_linux.run

   (3)cuda  环境变量配置:
              .1.     安装vim包

sudo apt install vim         # version 2:8.1.2269-1ubuntu5.4, or
sudo apt install vim-tiny    # version 2:8.1.2269-1ubuntu5.4
sudo apt install neovim      # version 0.4.3-3
sudo apt install vim-athena  # version 2:8.1.2269-1ubuntu5.4
sudo apt install vim-gtk3    # version 2:8.1.2269-1ubuntu5.4
sudo apt install vim-nox     # version 2:8.1.2269-1ubuntu5.4

             .2. 设置环境变量    gedit ~/.bashrc 推荐用这个,打开是文件好操作

vim ~/.bashrc 

打开文件,把以下的输入进去:(注意:   i插入    :wq 退出

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

(4)  检查安装是否成功

nvcc -V

三、 安装cudnn程序

(1) cudnn下载

        cuDNN是GPU加速计算深层神经网络的库。首先去官网 cuDNN Archive | NVIDIA Developer(2) 安装

    解压缩->在压缩后的cuda文件夹,打开终端

#注意,解压后的文件夹      将对应文件复制到 /usr/local中的cuda内 

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

猜你喜欢

转载自blog.csdn.net/m0_48919875/article/details/125161788