mirror of https://github.com/skygpu/skynet.git
Separate cuda build in a third step
parent
6453240f3a
commit
b922232840
|
@ -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
|
||||
|
|
|
@ -1,7 +1,5 @@
|
|||
scipy
|
||||
triton
|
||||
basicsr
|
||||
realesrgan
|
||||
accelerate
|
||||
transformers
|
||||
huggingface_hub
|
||||
|
|
|
@ -0,0 +1,2 @@
|
|||
basicsr
|
||||
realesrgan
|
Loading…
Reference in New Issue