记录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