Jagrut Thakare commited on
Commit
75b7732
·
1 Parent(s): eb74fcd
Files changed (1) hide show
  1. 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-cu112 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 . .
 
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 . .