added more files
This commit is contained in:
29
docker/pycudanode.dockerfile
Normal file
29
docker/pycudanode.dockerfile
Normal file
@ -0,0 +1,29 @@
|
||||
# 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
|
Reference in New Issue
Block a user