adityabalu/TorchMPISingularity:torch

$ singularity pull shub://adityabalu/TorchMPISingularity:torch

Singularity Recipe

bootstrap: shub
From: adityabalu/TorchMPISingularity:latest

%labels

MAINTAINER baditya@iastate.edu

%post
      # load environment variables
    . /environment

    # use bash as default shell
    echo "\n #Using bash as default shell \n" >> /environment
    echo 'SHELL=/bin/bash' >> /environment

    # make environment file executable
    chmod +x /environment

    # default mount paths
    which python
    pip install mpi4py
    pip install torch torchvision
    #git clone --recursive -b v1.0.0 https://github.com/pytorch/pytorch
    #CMAKE_PREFIX_PATH=/opt/conda/bin/
    #cd pytorch
    #git checkout v1.0.0
    #git submodule update --recursive
    #python setup.py install

Collection

• Name: adityabalu/TorchMPISingularity
• License: None

Metrics

Feedback

Was this page helpful?

Glad to hear it! Please tell us how we can improve.

Sorry to hear that. Please tell us how we can improve.