shengwenliang/pytorch_graph repository overview

This repository is created to run the pytorch_geometric project. The docker images support cuda-9.0 and already have been tested on my machine with ubuntu 14.04 and NVIDIA 1080ti card. Before using this repository, please install nvidia-docker2 to your machine.

Run Command: docker run --rm -it --init --runtime=nvidia --ipc=host --network=host --volume=$PWD:/app -e NVIDIA_VISIBLE_DEVICES=0 shengwenliang/pytorch_graph /bin/bash

this command allows the user to utilize the one GPU card on your machine. and if you want to use all GPU cards. please use "NVIDIA_VISIBLE_DEVICES=all" to replace "NVIDIA_VISIBLE_DEVICES=0".

Note, this image does not contain the pytorch_geometric project. you can download from GitHub project. When you using above the command, the container could link the internet. And your local file will appear at the /app folder of the container. So, you also could download the project to your local folder outside the container. In this case, you can enter the /app folder of the container to run the project.

If you has any question, please free contact to me by email: [email protected]⁠