Skip to content

Instantly share code, notes, and snippets.

@radiradev

radiradev/Dockerfile.ME.DepthWise

Created May 24, 2023 13:09
Show Gist options
  • Save radiradev/7af53500133773948c813d3e133e62c8 to your computer and use it in GitHub Desktop.
Save radiradev/7af53500133773948c813d3e133e62c8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Dockerfile.ME.DepthWise
FROM deeplearnphysics/larcv2:ub20.04-cuda11.6-pytorch1.13-larndsim
# Install ninja
RUN pip install ninja
ENV TORCH_CUDA_ARCH_LIST="6.0 7.0 7.5 8.0"
# Install MinkowskiEngine with DepthwiseConvolution from GitHub
RUN pip install -U --target=/usr/local/lib/python3.8/dist-packages \
git+https://github.com/fededagos/MinkowskiEngine -v --no-deps \
--install-option="--force_cuda" \
--install-option="--blas=openblas"
CMD ["bash"]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment