FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime

# Install pip dependencies
RUN pip install --no-cache-dir \
                'azureml-mlflow==1.39.0.post1' \
                'mlflow-skinny==1.26.1' \
                'func_to_script==0.1.0' \
                'matplotlib==3.5.3' \
                'datasets==2.6.' \
                'transformers==4.24.'\
                'evaluate==0.3.0' \
                'scipy==1.9.3' \
                'scikit-learn==1.1.3'

