diff --git a/Dockerfile.runtime+cuda b/Dockerfile.runtime+cuda index f8e9e93..32d3c4a 100644 --- a/Dockerfile.runtime+cuda +++ b/Dockerfile.runtime+cuda @@ -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 diff --git a/requirements.cuda.0.txt b/requirements.cuda.0.txt index e0f47c3..e31de88 100644 --- a/requirements.cuda.0.txt +++ b/requirements.cuda.0.txt @@ -1,7 +1,5 @@ scipy triton -basicsr -realesrgan accelerate transformers huggingface_hub diff --git a/requirements.cuda.2.txt b/requirements.cuda.2.txt new file mode 100644 index 0000000..4d3fee4 --- /dev/null +++ b/requirements.cuda.2.txt @@ -0,0 +1,2 @@ +basicsr +realesrgan