Skip to content

Instantly share code, notes, and snippets.

View dmargala's full-sized avatar

Daniel Margala dmargala

  • Berkeley Lab (NERSC)
View GitHub Profile
@dmargala
dmargala / Dockerfile
Created February 26, 2021 20:06
Dockerfile with CUDA 11.0 + UCX + OpenMPI + Python (numpy, numba, cupy, mpi4py) for use with Shifter at NERSC on Cori GPU
FROM nvcr.io/nvidia/cuda:11.0-devel-ubuntu18.04
# GNU compiler
RUN apt-get update -y && \
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
g++ \
gcc \
gfortran && \
rm -rf /var/lib/apt/lists/*