2020-05-12 12:12:06 +03:00
|
|
|
FROM mcr.microsoft.com/azureml/base-gpu:openmpi3.1.2-cuda10.1-cudnn7-ubuntu18.04
|
|
|
|
|
|
2020-09-10 11:34:34 +03:00
|
|
|
RUN python3 -m pip install --upgrade --no-cache-dir numpy scipy docopt dpu-utils more-itertools typing_extensions sentencepiece azureml-sdk pyyaml dill jellyfish
|
|
|
|
|
RUN pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
|
2020-05-12 12:12:06 +03:00
|
|
|
# Install torch scatter
|
2020-09-10 11:34:34 +03:00
|
|
|
RUN pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
|
