深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03

深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03

其他环境配置:https://blog.csdn.net/qq_26625581/article/details/109703131

安装显卡驱动

安装Nvidia显卡驱动,直接用图形界面安装:
深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03_第1张图片然后禁用集显驱动(nouveau):

sudo gedit /etc/modprobe.d/blacklist.conf

blacklist nouveau
options nouveau modeset=0

然后执行:

sudo update-initramfs -u

重启后,检查一下nouveau是否还在运行:

lsmod | grep nouveau  # 没输出代表禁用生效

最后检查一些独显驱动是否生效:

nvidia-smi

深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03_第2张图片

安装cuda

先查找对应显卡驱动的cuda版本:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

下载对应的版本:
https://developer.nvidia.com/cuda-toolkit-archive

选择runfile下载,在希望下载的目录运行网页提供的两条命令:深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03_第3张图片
选择Continue:
深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03_第4张图片输入accept:
深度学习环境搭建:Ubuntu20.04+显卡驱动450.80+cuda11.03_第5张图片去掉显卡选项后Install

配置环境变量

输入gedit ~/.bashrc命令打开文件,在文件结尾输入以下语句,保存。

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新环境变量

source ~/.bashrc

验证cuda是否安装成功

cd ~/NVIDIA_CUDA-11.0_Samples
make ./1_Utilities/deviceQuery/deviceQuery
make ./1_Utilities/bandwidthTest/bandwidthTest

最后运行一下这两个文件,都显示Result=PASS,说明CUDA安装成功。

你可能感兴趣的:(深度学习)