本文共 789 字，大约阅读时间需要 2 分钟。

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

其他环境配置：

安装显卡驱动

安装Nvidia显卡驱动，直接用图形界面安装:

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

blacklist nouveau

options nouveau modeset=0

然后执行：

sudo update-initramfs -u

重启后，检查一下nouveau是否还在运行：

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

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

nvidia-smi

安装cuda

先查找对应显卡驱动的cuda版本：

下载对应的版本：

选择runfile下载，在希望下载的目录运行网页提供的两条命令：

配置环境变量

输入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安装成功。

转载地址：http://rfbnn.baihongyu.com/