# Copyright (c) 2023, BLUEWALM. All rights reserved. 

ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:23.10-py3
FROM ${FROM_IMAGE_NAME}

WORKDIR /workspace/
ADD . /workspace/temp/
RUN cd temp; pip install pytorch_embeddings*.whl; cd ..; rm -r temp