Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions tools/docker/dev_container.Dockerfile
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
#syntax=docker/dockerfile:1.1.5-experimental
FROM tensorflow/tensorflow:2.1.0-custom-op-ubuntu16 as dev_container_cpu
FROM gcr.io/tensorflow-testing/nosla-cuda11.2-cudnn8.1-ubuntu18.04-manylinux2010-multipython as dev_container_cpu
ARG TF_PACKAGE
ARG TF_VERSION

# Temporary until custom-op container is updated
RUN ln -sf /usr/bin/python3 /usr/bin/python
RUN ln -sf /usr/local/bin/pip3 /usr/local/bin/pip
RUN ln -sf /usr/local/bin/python3.8 /usr/bin/python
RUN ln -sf /usr/local/bin/pip3.8 /usr/local/bin/pip
RUN pip install --default-timeout=1000 $TF_PACKAGE==$TF_VERSION

COPY tools/install_deps /install_deps
Expand Down