Skip to content
Branch: master
Find file History
Latest commit d584248 Apr 6, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
Dockerfile update dockerfile Apr 6, 2019 typo Apr 1, 2019


You can run PyTorch Geometric with CUDA 9.0 inside a docker image. The creation of our dockerfile refers to the dockerfiles provided by NVIDIA and PyTorch.

  1. Download the dockerfile to your host server.
  2. $ docker build -t "custom image name"
  3. $ docker run --rm -it --init --runtime=nvidia --ipc=host --network=host --volume=$PWD:/app -e NVIDIA_VISIBLE_DEVICES=0 "custom image name" /bin/bash

If you encounter any problems, please feel free to contact

You can’t perform that action at this time.