-
Notifications
You must be signed in to change notification settings - Fork 613
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Initial build against TF2.12 * support 3.11 --------- Co-authored-by: bhack <[email protected]>
- Loading branch information
1 parent
c35cde6
commit 9c1642a
Showing
23 changed files
with
71 additions
and
92 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -185,14 +185,14 @@ def configure_cuda(): | |
"CUDNN_INSTALL_PATH", | ||
os.getenv("CUDNN_INSTALL_PATH", "/usr/lib/x86_64-linux-gnu"), | ||
) | ||
write_action_env("TF_CUDA_VERSION", os.getenv("TF_CUDA_VERSION", "11.2")) | ||
write_action_env("TF_CUDA_VERSION", os.getenv("TF_CUDA_VERSION", "11.8")) | ||
write_action_env("TF_CUDNN_VERSION", os.getenv("TF_CUDNN_VERSION", "8")) | ||
|
||
write("test --config=cuda") | ||
write("build --config=cuda") | ||
write("build:cuda --define=using_cuda=true --define=using_cuda_nvcc=true") | ||
write( | ||
"build:cuda [email protected]_manylinux2014-cuda11.2-cudnn8.1-tensorrt7.2_config_cuda//crosstool:toolchain" | ||
"build:cuda [email protected]_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda//crosstool:toolchain" | ||
) | ||
|
||
|
||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,11 +1,14 @@ | ||
#syntax=docker/dockerfile:1.1.5-experimental | ||
ARG PY_VERSION | ||
FROM tensorflow/build:2.11-python$PY_VERSION as base_install | ||
FROM tensorflow/build:2.12-python$PY_VERSION as base_install | ||
|
||
ENV TF_NEED_CUDA="1" | ||
ARG PY_VERSION | ||
ARG TF_VERSION | ||
|
||
# TODO: Temporary due to build bug https://github.com/pypa/pip/issues/11770 | ||
RUN python -m pip install pip==22.3.1 | ||
|
||
# TODO: Remove this if tensorflow/build container removes their keras-nightly install | ||
# https://github.com/tensorflow/build/issues/78 | ||
RUN python -m pip uninstall -y keras-nightly | ||
|
@@ -42,7 +45,7 @@ RUN bazel build \ | |
--noshow_loading_progress \ | ||
--verbose_failures \ | ||
--test_output=errors \ | ||
[email protected]_manylinux2014-cuda11.2-cudnn8.1-tensorrt7.2_config_cuda//crosstool:toolchain \ | ||
[email protected]_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda//crosstool:toolchain \ | ||
build_pip_pkg && \ | ||
# Package Whl | ||
bazel-bin/build_pip_pkg artifacts $NIGHTLY_FLAG | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.