nelsonihc/singularity:torch-1804

$ singularity pull shub://nelsonihc/singularity:torch-1804

Singularity Recipe

BootStrap: shub
From: nelsonihc/singularity:base1804

%environment

%post
    . /opt/conda/etc/profile.d/conda.sh
    conda activate

    conda install -y pytorch cuda92 -c pytorch

    pip install \
        augmentor \
        PyHamcrest \
        pretrainedmodels \
        PyYAML \
        pycocotools \
        tensorboard \
        tensorboardX \
        tqdm \
        pybind11 \
        cppimport

    ldconfig /usr/local/cuda/lib64/stubs

    cd /tmp && \
    git clone https://github.com/uber/horovod.git && \
    cd horovod && HOROVOD_GPU_ALLREDUCE=NCCL python setup.py install --prefix=/opt/conda && \
    rm -r /tmp/horovod

    cd /tmp && \
    git clone https://github.com/pytorch/ignite.git && \
    cd ignite && python setup.py install --prefix=/opt/conda && \
    rm -r /tmp/ignite

    ldconfig

    # extra pips

Collection


View on Datalad

Metrics

key value
id /containers/nelsonihc-singularity-torch-1804
collection name nelsonihc/singularity
branch master
tag torch-1804
commit 339db7ad07d1c13c798bc6a9dc4274dfb9733420
version (container hash) f2c73428ed4515836368c634e5559277
build date 2019-02-13T20:23:56.833Z
size (MB) 8382
size (bytes) 4320624671
SIF Download URL (please use pull with shub://)
Datalad URL View on Datalad
Singularity Recipe Singularity Recipe on Datalad
We cannot guarantee that all containers will still exist on GitHub.