FROM pytorch/pytorch:latest

# Install Jupyter Notebook
RUN pip install jupyterlab

# Install additional packages
RUN pip install tensorflow

# Set environment variables, etc.
#ENV MY_ENV_VAR=myvalue