Separate cuda build in a third step

pull/4/head
Guillermo Rodriguez 2023-01-16 00:25:16 -03:00
parent 6453240f3a
commit b922232840
No known key found for this signature in database
GPG Key ID: EC3AB66D5D83B392
3 changed files with 4 additions and 3 deletions

View File

@ -11,8 +11,9 @@ workdir /skynet
copy requirements.cuda* ./ copy requirements.cuda* ./
run pip install -U pip ninja 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.1.txt
run pip install -v -r requirements.cuda.2.txt
copy requirements.test.txt requirements.test.txt copy requirements.test.txt requirements.test.txt
copy requirements.txt requirements.txt copy requirements.txt requirements.txt

View File

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

View File

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