NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.43k stars 360 forks source link

Can't install Minkowski Engine on Docker #450

Open thiagodma opened 2 years ago

thiagodma commented 2 years ago

I've been trying to install Minkowski Engine using a Dockerfile for some time now unsuccessfully.

Here's my dockerfile:

FROM pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel

##############################################
# You should modify this to match your GPU compute capability
# https://developer.nvidia.com/cuda-gpus#compute
# I have a RTX 3070 GPU
ENV TORCH_CUDA_ARCH_LIST="8.6"
##############################################

ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
ENV CUDA_HOME=/usr/local/cuda-11.1

# Install dependencies
RUN apt-get update
RUN apt-get install -y git ninja-build cmake build-essential libopenblas-dev \
    xterm xauth openssh-server tmux wget mate-desktop-environment-core

RUN apt-get clean
RUN rm -rf /var/lib/apt/lists/*

RUN pip install -U git+https://github.com/NVIDIA/MinkowskiEngine -v --no-deps --install-option="--force_cuda" --install-option="--blas=openblas"

Here is the gist with the logs.

KatherineJames commented 2 months ago

Hi @thiagodma - did you ever find a solution to this?