Spaces:
Running
Running
Jagrut Thakare
commited on
Commit
·
75b7732
1
Parent(s):
eb74fcd
v51
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -35,7 +35,7 @@ RUN pip cache purge
|
|
35 |
RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
|
36 |
RUN python3 -m pip cache purge
|
37 |
|
38 |
-
RUN pip install --user --no-cache-dir mxnet
|
39 |
RUN python3 -m pip cache purge
|
40 |
|
41 |
COPY --chown=user:user . .
|
|
|
35 |
RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
|
36 |
RUN python3 -m pip cache purge
|
37 |
|
38 |
+
RUN pip install --user --no-cache-dir mxnet onnxruntime-gpu==1.12 Cython requests==2.25.1 insightface==0.2.1 kornia==0.5.4 "numpy<=2.0" dill opencv-python onnx scikit-image wandb
|
39 |
RUN python3 -m pip cache purge
|
40 |
|
41 |
COPY --chown=user:user . .
|