101 次阅读

Ubuntu 配置 GPU 环境

注意如果装的是虚拟机则无法使用显卡,无法安装驱动。

测试环境

操作记录

Ubuntu 下离线安装包后缀为 .deb,安装 .deb 包使用命令

sudo dpkg -i 包名.deb

如果是其它扩展名或者没有扩展名的有可能是脚本程序

sudo bash 脚本名

安装显卡驱动

NVIDIA 驱动下载地址:https://www.nvidia.cn/geforce/drivers/ ,浏览器下载后默认保存路径为 ~/下载 (~/Downloads)

如果遇到这个错误就是说系统还没有安装 gcc,那么安装即可,安装后再重新执行驱动安装脚本

sudo apt update && sudo apt install -y build-essential

遇到询问是否安装 32 位兼容驱动,这个可以选 No,64 位系统没必要装 32 位的

安装完成

然后重启一下电脑加载显卡驱动,开始下面的操作

安装 CUDA

CUDA是一个由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

CUDA 下载地址:https://developer.nvidia.com/cuda-downloads

比如我这里是 Ubuntu 20.04 的电脑则选择,然后按照下面的说明操作就行

然后配置一下环境变量,否则 CUDA 带有的命令和相关的库在被需要时可能无法被找到

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc

echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

source ~/.bashrc

添加后,像 nvcc (CUDA 编译器)等就能直接在终端执行了

安装 cuDNN

cuDNN 是用于深度神经网络的 GPU 加速库。

下载地址:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b (需要登录会员帐号才能下载)

这个注册会员稍微有点麻烦,要填各种信息

注册完成会完成登录,之后就可以选择下载了

选择 cuDNN 要对应 CUDA 版本,如果是按照上面新安装的,则为最新版本,同样 cuDNN 选择最新的点开就行。或者是以前安装的版本,可以执行命令 nvidia-smi 查看显卡信息就能知道 CUDA 版本。

根据 CUDA 版本选定 cuDNN 版本后点开对应的,这里我就点开最上面那个来说明。我的系统是 Ubuntu 20.04,所以就下载下面标色的两个,一个是运行库,另一个是开发库,都要下载然后安装即可。

下载完后,因为是安装包,浏览器可能会怀疑不安全,询问是否保留,需要点保留

下载后先安装文件名不带 dev 的运行库,再安装 dev 开发库

发表评论