From 79892bc3ad9e2cf2bfdfa0881b667bc8006cc84a Mon Sep 17 00:00:00 2001 From: AbdBarho Date: Fri, 14 Oct 2022 15:50:51 +0200 Subject: [PATCH] run on bare metal --- .github/workflows/docker.yml | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index 07a774e..b14d407 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -4,10 +4,15 @@ on: [push] jobs: build: - runs-on: pytorch/manylinux-cuda113 + runs-on: ubuntu-latest + # container: + # image: pytorch/manylinux-cuda113 + env: + CU_VERSION: cu113 + PYTHON_VERSION: "3.10" + PYTORCH_VERSION: "1.12" steps: - run: | git clone https://github.com/facebookresearch/xformers /x cd /x - export CU_VERSION=cu113 PYTHON_VERSION=3.10 PYTORCH_VERSION=1.12 ./packaging/build_wheel.sh