452c18fc5d
Resolves #38247 PiperOrigin-RevId: 305114966 Change-Id: I83d00699783f9827ab526533fdeaaa3012086e66
11 lines
351 B
Docker
11 lines
351 B
Docker
# Options:
|
|
# tensorflow
|
|
# tensorflow-gpu
|
|
# tf-nightly
|
|
# tf-nightly-gpu
|
|
# Set --build-arg TF_PACKAGE_VERSION=1.11.0rc0 to install a specific version.
|
|
# Installs the latest version by default.
|
|
ARG TF_PACKAGE=tensorflow
|
|
ARG TF_PACKAGE_VERSION=
|
|
RUN python3 -m pip install --no-cache-dir ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
|