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


§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: 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?


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

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值