Fix dependencies

graph_neural_network/Dockerfile_mlcube

@@ -2,7 +2,7 @@ FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel
 
 WORKDIR /workspace/repository
 
-RUN pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
+# RUN pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
 RUN pip install scikit-learn==0.24.2
 RUN pip install torch_geometric==2.4.0
 RUN pip install --no-index  torch_scatter==2.1.1 torch_sparse==0.6.17 -f https://data.pyg.org/whl/torch-1.13.0+cu117.html

graph_neural_network/mlcube/readme.md

@@ -0,0 +1,74 @@
+# MLCube for Graph Neural Network
+
+MLCube™ GitHub [repository](https://github.com/mlcommons/mlcube). MLCube™ [wiki](https://mlcommons.github.io/mlcube/).
+
+## Project setup
+
+An important requirement is that you must have Docker installed.
+
+```bash
+# Create Python environment and install MLCube Docker runner 
+virtualenv -p python3 ./env && source ./env/bin/activate && pip install pip==24.0 && pip install mlcube-docker
+# Fetch the implementation from GitHub
+git clone https://github.com/mlcommons/training && cd ./training
+git fetch origin pull/762/head:feature/mlcube_graph_nn && git checkout feature/mlcube_graph_nn
+cd ./graph_neural_network/mlcube
+```
+
+Inside the mlcube directory run the following command to check implemented tasks.
+
+```shell
+mlcube describe
+```
+
+### MLCube tasks
+
+* Core tasks:
+
+Download dataset.
+
+```shell
+mlcube run --task=download_data -Pdocker.build_strategy=always
+```
+
+Process dataset.
+
+```shell
+mlcube run --task=process_data -Pdocker.build_strategy=always
+```
+
+Train GNN.
+
+```shell
+mlcube run --task=train -Pdocker.build_strategy=always
+```
+
+* Demo tasks:
+
+Download demo dataset.
+
+```shell
+mlcube run --task=download_demo -Pdocker.build_strategy=always
+```
+
+Run demo training.
+
+```shell
+mlcube run --task=demo -Pdocker.build_strategy=always
+```
+
+### Execute the complete pipeline
+
+You can execute the complete pipeline with one single command.
+
+* Core pipeline:
+
+```shell
+mlcube run --task=download_data,process_data,train -Pdocker.build_strategy=always
+```
+
+* Demo pipeline:
+
+```shell
+mlcube run --task=download_demo,demo -Pdocker.build_strategy=always
+```