Skip to content

pytorch-2.0.0-cpu #1139

pytorch-2.0.0-cpu

pytorch-2.0.0-cpu #1139

name: pytorch-2.0.0-cpu
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_cpu_build.yml
with:
remote_runtime_docker: bladedisc:latest-runtime-torch-2.0.0-cpu
develop_base_image: nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04
runtime_base_image: nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04
extra_build_args: --build-arg PYTHON_VERSION=PYTHON3.8 --build-arg ENABLE_FIND_FASTEST_APT_SOURCE=OFF
extra_envs: -e TORCH_BLADE_BUILD_WITH_CUDA_SUPPORT=OFF
-e TORCH_BLADE_CI_BUILD_TORCH_VERSION=2.0.0+cpu
exec_command: bash ./scripts/ci/test_pytorch_blade.sh
deploy_command: TORCH_VERSION=2.0.0+cpu bash ./scripts/ci/deploy_pytorch_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 }}