pytorch Container

Docker/Singularity image to run pytorch on Centos 6.9 kernel (Ubuntu 16.04) with GPU support.

If you have used this work for a publication, you must acknowledge SIH, e.g: "The authors acknowledge the technical assistance provided by the Sydney Informatics Hub, a Core Research Facility of the University of Sydney."

Quickstart for Artemis

Put this repo on Artemis e.g.

cd /project/<YOUR_PROJECT>
git clone https://github.com/Sydney-Informatics-Hub/pytorch-contained.git

Then cd pytorch-contained and modify the run_artemis.pbs script and launch with qsub run_artemis.pbs.

Otherwise here are the full instructions for getting there....

How to recreate

Build with docker

Check out this repo then build the Docker file.

sudo docker build . -t nbutter/pytorch:ubuntu1604

Run with docker.

To run this, mounting your current host directory in the container directory, at /project, and execute a run on the test images (that live in the container) run:

sudo docker run --gpus all -it -v `pwd`:/project nbutter/pytorch:ubuntu1604 /bin/bash -c "cd /project && python example_run.py"

Push to docker hub

sudo docker push nbutter/pytorch:ubuntu1604

See the repo at https://hub.docker.com/r/nbutter/pytorch

Build with singularity

export SINGULARITY_CACHEDIR=`pwd`
export SINGULARITY_TMPDIR=`pwd`

singularity build scarches.img docker://nbutter/pytorch:ubuntu1604

Run with singularity

To run the singularity image (noting singularity mounts the current folder by default)

singularity run --nv --bind /project:/project pytorch.img /bin/bash -c "cd "$PBS_O_WORKDIR" && python example_run.py"