OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Docker file for Cuda 11.4 built with Cudnn 8.0.5

Browse Source 
PiperOrigin-RevId: 403283584
Change-Id: I28ebe66bbedd1b4a820d0cac694e8a870f44d4b3
This commit is contained in:
Yash Katariya 2021-10-14 22:50:53 -07:00 committed by TensorFlower Gardener
parent cf42f72887
commit a0522d6847
2 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tensorflow/tools/ci_build/Dockerfile.rbe.cuda11.4-cudnn8.0.5-ubuntu18.04-manylinux2010-multipython Normal file
View File

@ -0,0 +1,25 @@
# Dockerfile to build a manylinux 2010 compliant cross-compiler.
#
# Builds a devtoolset gcc/libstdc++ that targets manylinux 2010 compatible
# glibc (2.12) and system libstdc++ (4.4).
#
# To push a new version, run:
# $ docker build -f Dockerfile.rbe.cuda11.4-cudnn8.0.5-ubuntu18.04-manylinux2010-multipython \
# --tag "gcr.io/tensorflow-testing/nosla-cuda11.4-cudnn8.0.5-ubuntu18.04-manylinux2010-multipython" .
# $ docker push gcr.io/tensorflow-testing/nosla-cuda11.4-cudnn8.0.5-ubuntu18.04-manylinux2010-multipython
FROM gcr.io/tensorflow-testing/nosla-cuda11.4-cudnn8.2-ubuntu18.04-manylinux2010-multipython
RUN apt-get update
RUN apt-get remove -y --allow-change-held-packages libcudnn8 libnccl2 libnccl-dev
RUN apt-get install -y --no-install-recommends --allow-downgrades --allow-change-held-packages \
libcublas-11-4 \
libcublas-dev-11-4 \
cuda-nvml-dev-11.4 \
cuda-command-line-tools-11.4 \
cuda-libraries-dev-11.4 \
cuda-minimal-build-11.4 \
libcudnn8=8.0.5.39-1+cuda11.1 \
libcudnn8-dev=8.0.5.39-1+cuda11.1
RUN rm -f /usr/local/cuda
RUN ln -s /usr/local/cuda-11.4 /usr/local/cuda

8
tensorflow/workspace3.bzl
View File

@ -17,11 +17,11 @@ def workspace():
http_archive(
name = "tf_toolchains",
sha256 = "8613ab9da262fd30d101550eab5780948db7471c7809cbac15a9e9a3143447f2",
strip_prefix = "toolchains-1.2.7",
sha256 = "214e306fc217bccf20e31ce32b943734b41b5d809fb84cf56b30093bef8f9249",
strip_prefix = "toolchains-1.2.9",
urls = [
"http://mirror.tensorflow.org/github.com/tensorflow/toolchains/archive/v1.2.7.tar.gz",
"https://github.com/tensorflow/toolchains/archive/v1.2.7.tar.gz",
"http://mirror.tensorflow.org/github.com/tensorflow/toolchains/archive/v1.2.9.tar.gz",
"https://github.com/tensorflow/toolchains/archive/v1.2.9.tar.gz",
],
)