Caleb Spradlin
initial commit
ab687e7
|
raw
history blame
1.01 kB

Requirements

pytorch-caney can be installed and used via anaconda environments and containers. A Docker container is provided and this same container can be converted to a Singularity container without loosing any functionalities.

CPU support is limited and the author does not provide any guarantee of usability.

Architecture

The container is built on top of NGC NVIDIA PYTORCH containers.

This application is powered by PyTorch and PyTorch Lighning AI/ML backends.

Example to Download the Container via Singularity

module load singularity
singularity build --sandbox pytorch-caney docker://nasanccs/pytorch-caney:latest

## Example to Install Anaconda Environment

``` bash
git clone [email protected]:nasa-nccs-hpda/pytorch-caney.git
cd pytorch-caney; conda env create -f requirements/environment_gpu.yml;
conda activate pytorch-caney

Container Usage

As an example, you can shell into the container:

singularity shell --nv -B <mounts> /path/to/container/pytorch-caney