3070使用tensorflow1.15或更低版本
Background
抢到了黄老板的3070,但是发现只有cuda11.1支持8.6算力,tf又没有支持cuda11.1的版本,故找到了如下解决方案。思路为套一层docker,以跳过对cuda与cudnn的检测。
环境
要求如下软件包:
- ubuntu 18(20可以,16未测试)
- Docker
- nvidia-docker2
大体步骤
- ubuntu禁用自带N卡驱动,安装nvidia官方驱动。截至11.18,nvidia对3070的linux驱动只有短期版本,这点需要注意。
- 安装cuda11.1,不用安装cudnn。cuda可能不用安装,但是安装了是测试通过了的。
- 安装docker与nvidia-docker2以获取docker对3070的支持。
- docker内拉取tensorflow官方镜像。这一步拉取一个ubuntu镜像配置环境应该也是可以的,没有测试。tensorflow官方镜像地址:https://hub.docker.com/r/tensorflow/tensorflow/ , 有详细介绍。运行时记得使用 --gpus all参数,否则不会使用GPU。