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" ]