# dobromirpopov/pycudanode

FROM python:slim

## System dependencies and Node.js/npm installation
RUN apt-get update && apt-get install -y \
    git curl nodejs npm gnupg2 wget build-essential

## Add NVIDIA package repositories
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
RUN dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN apt-get update

# Install CUDA and the accompanying libraries
RUN apt-get install -y cuda

# Clean up
RUN apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

# Setting up CUDA environmental variables
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64

## Upgrade pip
RUN pip install --upgrade pip

# Verify installations
RUN node --version && npm --version