mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2026-02-04 06:34:19 +01:00
29 lines
1,008 B
YAML
29 lines
1,008 B
YAML
name: Build Xformers
|
|
|
|
on: [push]
|
|
|
|
jobs:
|
|
build:
|
|
runs-on: ubuntu-latest
|
|
# container:
|
|
# image: pytorch/manylinux-cuda113
|
|
container:
|
|
image: python:3.10-slim
|
|
env:
|
|
DEBIAN_FRONTEND: noninteractive
|
|
XFORMERS_DISABLE_FLASH_ATTN: 1
|
|
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"
|
|
TORCH_CUDA_ARCH_LIST: "6.1"
|
|
NVCC_FLAGS: "--use_fast_math -DXFORMERS_MEM_EFF_ATTENTION_DISABLE_BACKWARD"
|
|
steps:
|
|
- run: |
|
|
apt-get update
|
|
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 cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y
|
|
pip install ninja
|
|
pip wheel --wheel-dir=/data git+https://github.com/facebookresearch/xformers.git@ba93c5012d00bd1b010514a7bc9bd938c1ad6149#egg=xformers
|