Separate cuda build in a third step

pull/25/head
Guillermo Rodriguez 2023-01-16 00:25:16 -03:00
parent bc00fe35c8
commit 43ed6efc75
3 changed files with 4 additions and 3 deletions

View File

@ -11,8 +11,9 @@ workdir /skynet
copy requirements.cuda* ./
run pip install -U pip ninja
run pip install -r requirements.cuda.0.txt
run pip install -v -r requirements.cuda.0.txt
run pip install -v -r requirements.cuda.1.txt
run pip install -v -r requirements.cuda.2.txt
copy requirements.test.txt requirements.test.txt
copy requirements.txt requirements.txt

View File

@ -1,7 +1,5 @@
scipy
triton
basicsr
realesrgan
accelerate
transformers
huggingface_hub

View File

@ -0,0 +1,2 @@
basicsr
realesrgan