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

ENV CUDA_VISIBLE_DEVICES=all

LABEL sct_gnn=sct_gnn

COPY . /root/workspace/sct-gnn/

RUN apt-get update

RUN apt -y install build-essential

RUN apt-get -y install git 

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

RUN pip3 install torch_geometric -U

RUN mkdir -p /root/workspace/data/ 

RUN mkdir -p /root/workspace/out/