3070使用tensorflow1.15或更低版本

3070使用tensorflow1.15或更低版本

Background

抢到了黄老板的3070,但是发现只有cuda11.1支持8.6算力,tf又没有支持cuda11.1的版本,故找到了如下解决方案。思路为套一层docker,以跳过对cuda与cudnn的检测。

环境

要求如下软件包:

  • ubuntu 18(20可以,16未测试)
  • Docker
  • nvidia-docker2

大体步骤

  1. ubuntu禁用自带N卡驱动,安装nvidia官方驱动。截至11.18,nvidia对3070的linux驱动只有短期版本,这点需要注意。
  2. 安装cuda11.1,不用安装cudnn。cuda可能不用安装,但是安装了是测试通过了的。
  3. 安装docker与nvidia-docker2以获取docker对3070的支持。
  4. docker内拉取tensorflow官方镜像。这一步拉取一个ubuntu镜像配置环境应该也是可以的,没有测试。tensorflow官方镜像地址:https://hub.docker.com/r/tensorflow/tensorflow/ , 有详细介绍。运行时记得使用 --gpus all参数,否则不会使用GPU。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值