tensorflow-GPU docker运行

1.tensorflow安装版本介绍

在这里插入图片描述

在服务器上部署的时候,可能宿主机直接用apt install安装的CUDA9.1(ubuntu18.04支持apt install 安装CUDA9.1),此时用pip install tensorflow–gpu 安装的 1.11.0 不能支持CUDA9.1。所以此时可以采用两种方案,一种是直接安装编译好的 tensorflow-gpu。由于已经在CUDA9.1上编译好,此时都可以支持CUDA9.1。下载地址见:https://github.com/mind/wheels/releases/

2.ubuntu安装cuda,cudnn,tensorflow-gpu

ubuntu18.04直接使用apt进行安装:

sudo aptitude install nvidia-cuda-toolkit --without-recommends

需要注意的是在ubuntu16以及自己安装的CUDA,安装位置位于:/usr/local下面。而现在安装位置在:/usr/lib/x86_64-linux-gnu$ cd下面。其中/usr/lib/cuda中可以查看cuda版本。

安装cudnn:

tar -zxf cudnn-9.1-linux-x64-v7.tgz
cd cuda
mkdir -p lib/x86_64-linux-gnu
mv lib64/* l
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值