GPUDockerfile实例：tensorflow-gpu

AI拉呱

于 2020-08-27 14:33:27 发布

阅读量404

点赞数

CC 4.0 BY-SA版权

分类专栏： linux实战从入门到精通

本文链接：https://blog.youkuaiyun.com/weixin_32393347/article/details/88648881

linux实战从入门到精通同时被 2 个专栏收录

67 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

Mac 从入门到熟练

50 篇文章

订阅专栏

这篇博客详细介绍了如何在Ubuntu 16.04环境下，基于Python 3构建一个TensorFlow Dockerfile，该Dockerfile将配合CUDA 9和CuDNN 7来优化GPU运算。

tensorflow 与cuda对应版本

ubuntu 1604.py3 Dockerfile

FROM scratch
ADD ubuntu-xenial-core-cloudimg-amd64-root.tar.gz /

# a few minor docker-specific tweaks
# see https://github.com/docker/docker/blob/9a9fc01af8fb5d98b8eec0740716226fadb3735c/contrib/mkimage/debootstrap
RUN set -xe \
	\
# https://github.com/docker/docker/blob/9a9fc01af8fb5d98b8eec0740716226fadb3735c/contrib/mkimage/debootstrap#L40-L48
	&& echo '#!/bin/sh' > /usr/sbin/policy-rc.d \
	&& echo 'exit 101' >> /usr/sbin/policy-rc.d \
	&& chmod +x /usr/sbin/policy-rc.d \
	\
# https://github.com/docker/docker/blob/9a9fc01af8fb5d98b8eec0740716226fadb3735c/contrib/mkimage/debootstrap#L54-L56
	&& dpkg-