A container for pytorch 1.5 and fastslow with cuda10.1 ubuntu16.04 python 3.6 base
1.1K
Docker/Singularity repo for pytorch and https://github.com/facebookresearch/SlowFast
If you have used this work, you must acknoledge SIH, e.g: "The authors acknowledge the technical assistance provided by the Sydney Informatics Hub, a Core Research Facility of the University of Sydney."
To get this repo and scripts or to build locally
git clone https://github.com/natbutter/artemis-pytorch.git
sudo docker build . -t pyslowfast
#or
sudo singularity build --writable pyslowfast.img pyslowfast.build
To run a Python script (e.g. testGPU.py) that is in your current directory on your host by mounting inside the container in the project directory and then executing with docker:
sudo docker run -v `pwd`:/project nbutter/pyfastslow /bin/bash -c "cd /project && python testGPU.py"
To run with singularity in a PBS jobscript:
#!/bin/bash
#PBS -P <yourproject>
#PBS -N myrun
#PBS -l select=1:ncpus=1:mem=1gb:ngpus=1
#PBS -l walltime=00:01:00
#PBS -q defaultQ
module load singularity
cd $PBS_O_WORKDIR
#Mount the current directory in the container at /project, change directories and run your script
singularity exec -B `pwd`:/project ../pyslowfast.img /bin/bash -c "cd /project && python testGPU.py"
Content type
Image
Digest
Size
4.1 GB
Last updated
about 6 years ago
docker pull nbutter/pyfastslow