From 170a742cac72939d06c2dad2678a47fea427c75b Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Fri, 14 Oct 2022 16:10:26 +0200
Subject: [PATCH] g

---
 .github/workflows/docker.yml | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 3dd3a56..77e2629 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -8,16 +8,14 @@ jobs:
     # container:
     #   image: pytorch/manylinux-cuda113
     container:
-      image: python:3.9-slim
+      image: python:3.10-slim
       env:
-        CU_VERSION: cu113
-        PYTHON_VERSION: "3.9"
-        PYTORCH_VERSION: "1.12"
         DEBIAN_FRONTEND: noninteractive
     steps:
       - run: |
+          apt-get install gpg wget -y
+          wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
+          dpkg -i cuda-keyring_1.0-1_all.deb
           apt-get update
-          apt-get install git -y
-          git clone https://github.com/facebookresearch/xformers x
-          cd x
-          ./packaging/build_wheel.sh
+          apt-get install cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y
+          FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0;8.6+PTX" pip wheel --wheel-dir=/data git+https://github.com/facebookresearch/xformers.git@ba93c5012d00bd1b010514a7bc9bd938c1ad6149#egg=xformers