Skip to content

add tf28+cu113 CI job #777

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Yancey0623 wants to merge 8 commits into
base: main
Choose a base branch
from
Open

add tf28+cu113 CI job #777

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
241a699
add tf28 with cu113
Nov 23, 2022
3820042
use py38
Yancey0623 Nov 23, 2022
705300e
update
Yancey0623 Nov 24, 2022
db7fce8
update
Yancey0623 Nov 30, 2022
6474e25
fix tf blade
Yancey0623 Nov 30, 2022
e05eeb9
install protobuf==3.20.1
Yancey0623 Nov 30, 2022
c00ce7e
update
Yancey0623 Nov 30, 2022
a76f9f5
fix ci
Yancey0623 Nov 30, 2022
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
28 changes: 28 additions & 0 deletions .github/workflows/tf28_gpu.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
name: tf2_8-cuda11_3
on:
push:
branches: [ main ]
pull_request:
branches: [ main ]
schedule:
- cron: '0 17 * * *' # 1:00 am UTC+8:00
workflow_dispatch:

jobs:
build:
uses: ./.github/workflows/resuable_gpu_build.yml
with:
cuda_version: cu113
runner_tag: gpu-a10
remote_runtime_docker: bladedisc:latest-runtime-tensorflow2.8
develop_base_image: nvidia/cuda:11.3.0-cudnn8-devel-ubuntu18.04
runtime_base_image: tensorflow/tensorflow:2.8.0-gpu
extra_build_args: --build-arg PYTHON_VERSION=PYTHON3.8 --build-arg DISC_HOST_TF_VERSION="tensorflow-gpu==2.8" --build-arg ENABLE_FIND_FASTEST_APT_SOURCE=OFF
exec_command: bash ./scripts/ci/build_and_test.sh
tf_blade_command: bash ./scripts/ci/test_tensorflow_blade.sh cu113 2.8
deploy_command: bash ./scripts/ci/deploy_tensorflow_blade.sh
secrets:
ALIYUN_DOCKER_USERNAME: ${{ secrets.ALIYUN_DOCKER_USERNAME }}
ALIYUN_DOCKER_PASSWORD: ${{ secrets.ALIYUN_DOCKER_PASSWORD }}
DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }}
DOCKER_PASSWORD: ${{ secrets.DOCKER_PASSWORD }}
3 changes: 3 additions & 0 deletions docker/scripts/install-python.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,9 @@ function install_venv() {
python3 -m virtualenv ${DISC_VENV}
source ${DISC_VENV}/bin/activate
if [[ ! -z "${DISC_HOST_TF_VERSION}" ]]; then
if [[ "${DISC_HOST_TF_VERSION}" == "tensorflow-gpu==2.8" ]]; then
pip install -q protobuf==3.20.1
fi
echo "install TensorFlow: "${DISC_HOST_TF_VERSION} "..."
pip install -q ${DISC_HOST_TF_VERSION}
fi
Expand Down
2 changes: 1 addition & 1 deletion tensorflow_blade/requirement-common.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ parameterized==0.8.1
cython==0.29.23

# for linting
black==19.10b0
black==22.3.0
flake8==3.9.2
flake8-isort==4.0.0
isort==5.8.0
Expand Down
3 changes: 3 additions & 0 deletions tensorflow_blade/requirement-tf2.8-cu113.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
-r requirement-common.txt
protobuf==3.20.1
tensorflow-gpu==2.8.0