Skip to content
Permalink
Browse files

Update README.md

  • Loading branch information
shengwenLeong committed Mar 8, 2020
1 parent 23327b3 commit 1780704df03faba696b0d5e0875a3bd68cffb6fa
Showing with 3 additions and 3 deletions.
  1. +3 −3 docker/README.md
@@ -1,7 +1,7 @@
# Docker

You can run PyTorch Geometric with CUDA 9.0 inside a docker image.
The creation of [our dockerfile](https://github.com/rusty1s/pytorch_geometric/blob/master/docker/Dockerfile) refers to the dockerfiles provided by [NVIDIA](https://gitlab.com/nvidia/cuda/tree/ubuntu16.04) and [PyTorch](https://github.com/anibali/docker-pytorch).
You can run PyTorch Geometric with CUDA 10.1 inside a docker image.
The creation of [our dockerfile](https://github.com/rusty1s/pytorch_geometric/blob/master/docker/Dockerfile) refers to the dockerfiles provided by [NVIDIA](https://gitlab.com/nvidia/cuda/tree/ubuntu18.04) and [PyTorch](https://github.com/anibali/docker-pytorch).

1. Download the dockerfile to your host server.
2. `$ docker build -t "custom image name"`
@@ -49,4 +49,4 @@ to run on the CPU, or
```
singularity exec --nv geometric.sif python3 gat.py
```
to run on the GPU.
to run on the GPU.

0 comments on commit 1780704

Please sign in to comment.
You can’t perform that action at this time.