FROM python:3.8-slim

FROM nvidia/cuda:11.1.1-base-ubuntu18.04

FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime

ENV NVIDIA_VISIBLE_DEVICES all

ENV NVIDIA_DRIVER_CAPABILITIES compute,utility

RUN apt-get update

RUN apt -y install build-essential

RUN apt-get -y install git 

RUN pip3 install torch-geometric==2.0.0 torch_sparse==0.6.12 torch_scatter==2.0.9 -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.html

LABEL mignn=mignn

COPY . /root/workspace/

ENV CUDA_VISIBLE_DEVICES=all

ENV PORT=8881

EXPOSE 8881