Skip to content

CMU-INF-DIVA/actev_base

Repository files navigation

ActEV Base Environment

Docker Image

Author: Lijun Yu

Email: lijun@lj-y.com

Base image for ActEV SDL submission of CMU.

Environment

Usage

Docker Image

Github Packages

docker pull ghcr.io/cmu-inf-diva/actev_base

# Test it
docker run -it --rm --gpus all --ipc host ghcr.io/cmu-inf-diva/actev_base \
    bash -ic "python -c 'import torch; assert torch.cuda.is_available()'; nvidia-smi"

Container directory structure

  • /app - Work directory
    • env - Python environment
    • core - System implementation
    • data - Mount point for data

Local Development

Install miniconda, then run

./setup.sh [<name>] # name defaults to `actev`

If you want to patch PyTorch with this PR, run

conda activate <name>
./patch_pytorch.sh