29 lines
1007 B
Docker
29 lines
1007 B
Docker
# 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 |