Skip to content

Commit c2f9519

Browse files
authored
Adjust package paths for torch, numpy, numba in Torch Docker image (#751)
1 parent 3688d9e commit c2f9519

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

docker/dockerfile.torch

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ RUN pip install --no-cache-dir --no-deps torch torchmetrics \
2929
COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/pytorch backends/pytorch
3030

3131
# DLFW Python packages
32-
COPY --chown=1000:1000 --from=dlfw /opt/conda/lib/python3.8/site-packages/numba /usr/local/lib/python3.8/dist-packages/numba
33-
COPY --chown=1000:1000 --from=dlfw /opt/conda/lib/python3.8/site-packages/numpy /usr/local/lib/python3.8/dist-packages/numpy
34-
COPY --chown=1000:1000 --from=dlfw /opt/conda/lib/python3.8/site-packages/torch /usr/local/lib/python3.8/dist-packages/torch
32+
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/numba /usr/local/lib/python3.8/dist-packages/numba
33+
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/numpy /usr/local/lib/python3.8/dist-packages/numpy
34+
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/torch /usr/local/lib/python3.8/dist-packages/torch
3535

3636
RUN pip install --no-deps fastai fastcore fastprogress fastdownload
3737
RUN pip install matplotlib

0 commit comments

Comments
 (0)