Dockerfile.ci_gpu 257 Bytes
Newer Older
1
2
3
4
5
6
7
8
From dgllib/dgl-sagemaker-gpu:dgl_0.4_pytorch_1.2.0_rdkit

RUN pip uninstall -y scikit-learn
RUN pip install scikit-learn==0.22.2.post1
RUN pip uninstall -y pandas
RUN pip install pandas==0.24.2
RUN pip uninstall -y dgl-cu100
RUN pip install --pre dgl-cu101