记录ubuntu服务器加装RTX30系列显卡搭建tensorflow环境的过程
搭建深度学习环境,需要使用到的深度学习框架和显卡之间存在对应关系,但是很多网上找到的深度学习算法实现是基于低版本的框架代码实现的,由于tensorflow框架版本变动较大,算法工程师要求安装1.4版本的tensorflow,采购显卡时未考虑到兼容性问题,网上找了好多帖子,这里记录一下完整的过程。
具体的安装步骤分为:
- 安装cuda,cudnn
- 安装docker,nvidia docker2
- 拉取nvidia docker runtime,tensorflow 1.4镜像
- 启动tensorflow容器
1 安装cuda cudnn
RTX 30系列显卡支持的显卡驱动最低版本为 455.38,支持的CUDA版本是11.x,
具体配置为
NVIDIA-SMI 455.38 Driver Version: 455.38 CUDA Version: 11.1
RTX30系列显卡安装指导
https://blog.csdn.net/baidu_26678247/article/details/109727892
2 安装docker,nvidia docker2
这里需要注意的是,安装docker-ce和nvidia-docker2是有版本匹配要求的,如果发现安装完成命令也可以执行但是遇到下面的错误可能是镜像版本不

本文详细记录了在Ubuntu18.04服务器上,如何加装RTX30系列显卡,并搭建Tensorflow1.4环境的过程,包括安装CUDA和cuDNN,部署Docker与nvidia-docker2,以及启动Tensorflow容器的步骤。过程中解决了版本匹配和运行时问题。

5896

被折叠的 条评论
为什么被折叠?



