From 65a8c12bb42d855ed341a0af25cee754c30f728d Mon Sep 17 00:00:00 2001 From: Jonas Frey Date: Wed, 12 Aug 2020 13:04:41 +0200 Subject: [PATCH] Fix Docker build for CUDA version. --- Dockerfile-GPU | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Dockerfile-GPU b/Dockerfile-GPU index 1fb71b1..2af0eb1 100644 --- a/Dockerfile-GPU +++ b/Dockerfile-GPU @@ -1,12 +1,12 @@ # Install NVIDIA GPU image -FROM nvidia/cuda:9.2-cudnn7-devel +FROM nvidia/cuda:9.2-cudnn7-devel-ubuntu18.04 # Install dependencies -RUN apt-get update && apt-get install -y --no-install-recommends python3-dev python3-pip python3-nose python3-setuptools libblas-dev liblapack-dev cmake ffmpeg gfortran git && rm -rf /var/lib/apt/lists/* +RUN apt-get update && apt-get install -y --no-install-recommends python3-dev python3-pip python3-nose python3-setuptools libblas-dev liblapack-dev cmake ffmpeg gfortran git llvm-8-dev && rm -rf /var/lib/apt/lists/* # Install required Python packages RUN pip3 install --upgrade pip -RUN pip3 install numpy scipy cython librosa future +RUN LLVM_CONFIG=llvm-config-8 pip3 install numpy scipy cython librosa future # Obtain libgpuarray & pygpu. RUN git clone https://github.com/Theano/libgpuarray.git @@ -44,4 +44,4 @@ COPY . ./ ADD https://tuc.cloud/index.php/s/m9smX4FkqmJaxLW/download ./model/BirdNET_Soundscape_Model.pkl # Add entry point to run the script -ENTRYPOINT [ "python3", "./analyze.py" ] \ No newline at end of file +ENTRYPOINT [ "python3", "./analyze.py" ]