当前位置: 代码迷 >> 综合 >> Ubuntu20.04安装CUDA11.0+cuDNN7.6.5教程
  详细解决方案

Ubuntu20.04安装CUDA11.0+cuDNN7.6.5教程

热度:95   发布时间:2023-11-17 14:58:09.0

前言

自己总结的从新安装cuda+cudnn+anaconda+pycharm,于此写一个教程

安装NVIDIA驱动

在安装好Ubuntu20.04系统后,首先Ctrl+Alt+T进入终端,执行命令sudo apt-get update更新软件源列表,接着执行命令sudo add-apt-repository ppa:graphics-drivers添加ppa仓库,然后再执行sudo apt-get update。在添加好ppa仓库后,可使用sudo apt-cache search nvidia或者ubuntu-drivers devices命令查看可安装的NVIDIA驱动版本。其中后一条命令还会给出当前显卡推荐的显卡驱动版本。在安装驱动之前,如果已经安装过NVIDIA驱动,那么可使用命令sudo apt-get remove --purge nvidia*先行卸载。确定要安装的驱动程序版本后,即可使用命令sudo apt-get install nvidia-driver-470进行安装,470是我安装驱动的时候最新的版本。如果不想麻烦,也可以使用命令sudo ubuntu-drivers autoinstall自动安装推荐的NVIDIA驱动版本。在等待大约十分钟后安装完毕,执行命令sudo reboot重启电脑,进入终端执行命令nvidia-smi验证驱动是否安装成功。一般情况下安装是会成功的,但如果不成功也不要着急。重启后失败则Ctrl+Alt+F1进入终端,使用命令sudo apt-get remove --purge nvidia*卸载掉NVIDIA驱动再重新安装即可。
 

安装CUDA+cuDNN

在装好NVIDIA驱动后,即可安装CUDA+cuDNN。首先到CUDA官网去下载需要的CUDA版本,一般选择下载.run格式的本地安装包。我要安装的CUDA版本为11.0,对应的下载地址是CUDA11.0。

同时去cuDNN官网下载CUDA对应的cuDNN版本,下载期间会需要登录,但可以直接用QQ或者微信登录以节省注册的时间。CUDA11.0对应的cuDNN版本为8.0.5,下载地址为CUDA8.0.5,下载页面如下所示:点cuDNN Library for Linux[x86_64]

 下载好CUDA和cuDNN后,安装就比较简单了。首先安装CUDA,使用命令

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

 进行下载和安装唯一需要注意的是不要选择安装驱动的选项即可(在第一项按Enter取消),其它可都按照默认选项安装。安装完成后输入

sudo 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

保存即可完成环境变量的配置。CUDA安装完成后进入存放cuDNN的文件夹下,首先执行解压命令

tar -zxvf cudnn-11.0-linux-x64-v8.0.5.39.tgz

然后进入解压后的文件夹,复制里面的cuda/include/cudnn.h到/usr/local/cuda/include/文件夹下,以及复制cuda/lib64/下的所有文件到/usr/local/cuda/lib64/文件夹下,至此CUDA+cuDNN已全部安装完成。

安装Anaconda3

安装Anaconda3,首先到Anaconda3下载地址下载安装包,下载页面如下所示:

 下载完成后,打开终端进入安装包所在文件夹,执行命令sudo sh xxxxx.sh(你的文件的名字)进行安装,安装过程中直接使用默认值就好,在最后会提示是否添加Anaconda3的路径到.bashrc,选择yes软件会自动添加环境变量,最后安装完成。安装好Anaconda3后,首先需要做的事情是添加国内的软件源,这里推荐的是清华镜像源。添加的步骤为,打开终端,依次执行命令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --remove channels defaults
conda config --set show_channel_urls yes

即完成镜像源的添加,其中第三条命令移除了Anaconda3的默认下载通道,这是推荐的做法,因为如果软件选择它作为下载通道,那么下载将是很漫长痛苦的事情。镜像源配置好后,就可以执行命令conda create -n fucccc python=3.6创建自己的虚拟环境了,这里fucccc是需要创建的虚拟环境名,python=3.6.6用于指定安装的Python版本,执行后在Anaconda3的安装目录下的envs文件夹下可看到对应的虚拟环境名文件夹。在虚拟环境创建后,执行命令source activate fucccc切换到虚拟环境(要退出则执行source deactivate fucccc),然后执行命令conda install pip安装pip软件包下载工具,安装完成后记得执行命令python -m pip install --upgrade pip更新pip到最新版本。由于pip的国外源下载一般非常慢,在安装完之后也需要对pip进行国内软件源配置,即执行命令cd切换到用户文件夹下,接着执行命令mkdir .pip创建.pip文件夹,然后进入文件夹执行命令gedit pip.conf编辑文件,添加如下内容:

[global]
timeout = 6000
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host = https://pypi.tuna.tsinghua.edu.cn

conda常用指令

    1)conda list 查看安装了哪些包。
    2)conda env list 查看当前存在哪些虚拟环境
    3)conda update conda 检查更新当前conda

   4)conda remove -n your_env_name --all, 即可删除虚拟环境(要换成你的虚拟环境的名字)。

  相关解决方案