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

README.md

Docker

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 liangshengwen@ict.ac.cn.

You can’t perform that action at this time.