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* ./
|
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
|
||||||
|
|
|
@ -1,7 +1,5 @@
|
||||||
scipy
|
scipy
|
||||||
triton
|
triton
|
||||||
basicsr
|
|
||||||
realesrgan
|
|
||||||
accelerate
|
accelerate
|
||||||
transformers
|
transformers
|
||||||
huggingface_hub
|
huggingface_hub
|
||||||
|
|
|
@ -0,0 +1,2 @@
|
||||||
|
basicsr
|
||||||
|
realesrgan
|
Loading…
Reference in New Issue