FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime

RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
RUN conda install grpcio
RUN pip install \
    pytorch-lightning==1.1.3 \
    lightning-bolts==0.3.0 \
    sklearn \
    seaborn \
    pydevd-pycharm~=212.5712.39 \
    opencv-python
