[ GPU 环境搭建 ] 如何在云服务器上搭建 NIVIDA GPU 开发环境

Chen&Ru

已于 2024-06-13 12:42:21 修改

阅读量1k

点赞数 1

分类专栏： Engineering 文章标签： ubuntu 深度学习服务器

于 2022-05-09 09:01:33 首次发布

本文链接：https://blog.youkuaiyun.com/Bill_seven/article/details/124629038

版权

Engineering 专栏收录该内容

3 篇文章

订阅专栏

本文记录了在Ubuntu系统上搭建GPU开发环境的步骤，包括安装Nvidia驱动、CUDA和CUDNN。首先，介绍了系统版本和显卡型号的查看方法，然后详细阐述了如何安装显卡驱动，通过Nvidia官网找到对应版本并执行安装。接着，安装CUDA驱动，使用`ubuntu-drivers devices`检查支持的版本，并根据提示安装。最后，安装CUDNN，需要在Nvidia官网注册下载并按照指示进行。安装完成后，通过运行测试代码验证环境配置成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

§1. General Purposes: Why I wrote this?

最近因业务需要，公司购买了训练用的云服务器，自己搭建 GPU 开发环境环境的同时觉得顺道记录下比较好，就有了这篇文章。下面先 Post 一下购买的云服务器配置和系统版本：

系统版本

lsb_release -a

sysver

显卡型号

lshw -C display

nvidia

准备工作

apt install ubuntu-drivers-common software-properties-common nvidia-cuda-toolkit zlib1g

§2. Installation: How to install Drivers & CUDA & CUDNN?

显卡驱动

若显卡内核驱动已安装，可用下面指令查看显卡内核驱动版本：

cat /proc/driver/nvidia/version

kernel
若未安装显卡驱动，可至 Nvidia 官网查看 CUDA 驱动版本对应关系，并下载显卡对应的驱动：NVIDIA驱动下载页。

apt-key add /var/nvidia-driver-local-repo-ubuntu1804-510.47.03/7fa2af80.pub
dpkg -i nvidia-driver-local-repo-ubuntu1804-510.47.03_1.0-1_amd64.deb

CUDA

查看支持的 driver 版本并安装对应 CUDA 驱动：

ubuntu-drivers devices
apt-get update
apt-get install cuda-drivers-510=510.47.03-1

这个过程中如果出现版本不对报错等，可以直接按提示手动安装对应需要的版本。
到 Nvidia 官网下载对应的 CUDA Toolkit：CUDA Toolkit下载页。按照页面指导执行安装即可。（详细过程也可以参考官方安装指南）

CUDNN

安装 CUDNN 需要到官网注册下载：CUDNN 安装说明。具体安装操作请至官网查看。安装完毕后可以跑一下 mnist 的测试代码，通过会显示如下结果：

§3. Summary: What I have done?

大致就是把网上关于显卡开发环境安装部署的一堆信息整合到了一起吧。