#FROM python:3.8.6-slim
FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
#FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel

## The MAINTAINER instruction sets the author field of the generated images.
MAINTAINER hseki@ami.inc

## DO NOT EDIT the 3 lines.
RUN mkdir /physionet
COPY ./ /physionet
WORKDIR /physionet

## Install your dependencies here using apt install, etc.
RUN apt-get update && apt-get install -y \
    automake \
    build-essential \
    curl \
    git \
    emacs \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*

## Include the following line if you have a requirements.txt file.
RUN pip install -r requirements.txt
#RUN pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
