Commit 7e6d3ca6 authored by nanahira's avatar nanahira

blackwell

parent c32ab44f
Pipeline #40406 canceled with stages
in 2 minutes and 43 seconds
...@@ -38,16 +38,21 @@ build-xformers-8.6: ...@@ -38,16 +38,21 @@ build-xformers-8.6:
variables: variables:
GRAPHIC_VERSION: '8.6' # 30xx or A4000 A5000 GRAPHIC_VERSION: '8.6' # 30xx or A4000 A5000
#build-xformers-8.9: build-xformers-8.9:
# extends: .build-xformers extends: .build-xformers
# variables: variables:
# GRAPHIC_VERSION: '8.9' # 40xx GRAPHIC_VERSION: '8.9' # 40xx
build-xformers-6.1: build-xformers-6.1:
extends: .build-xformers extends: .build-xformers
variables: variables:
GRAPHIC_VERSION: '6.1' # 10xx GRAPHIC_VERSION: '6.1' # 10xx
build-xformers-12.0:
extends: .build-xformers
variables:
GRAPHIC_VERSION: '12.0' # 50xx
.deploy: .deploy:
stage: deploy stage: deploy
tags: tags:
......
...@@ -3,6 +3,6 @@ FROM ${BASE} ...@@ -3,6 +3,6 @@ FROM ${BASE}
ARG GRAPHIC_VERSION ARG GRAPHIC_VERSION
RUN env FORCE_CUDA=1 CUDA_VISIBLE_DEVICES=0 TORCH_CUDA_ARCH_LIST=${GRAPHIC_VERSION} \ RUN env FORCE_CUDA=1 CUDA_VISIBLE_DEVICES=0 TORCH_CUDA_ARCH_LIST=${GRAPHIC_VERSION} \
pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.git@v0.0.20#egg=xformers pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.git@v0.0.23.post1#egg=xformers
CMD ["--listen", "--port", "8080", "--xformers"] CMD ["--listen", "--port", "8080", "--xformers"]
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment