mirror of
https://github.com/bspeice/neuralnets-docker
synced 2024-12-04 20:58:09 -05:00
29 lines
765 B
Docker
29 lines
765 B
Docker
FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
|
|
MAINTAINER Bradlee Speice
|
|
|
|
LABEL Description="Jupyter server setup for ECBM E4040 Neural Networks" Version="0.2"
|
|
|
|
# Update our cache first
|
|
RUN apt-get update && \
|
|
|
|
# Don't install broken Pip
|
|
apt-get install -y python-pip=8.1.1-2 python3-pip=8.1.1-2 python-pip-whl=8.1.1-2 && \
|
|
|
|
# Install the Scipy stuff we need
|
|
apt-get install -y \
|
|
python3 libpython3-dev \
|
|
python-pandas python-matplotlib python-sklearn \
|
|
texlive-latex-extra texlive-fonts-recommended texlive-generic-recommended pandoc
|
|
|
|
|
|
# And the python-specific tools
|
|
RUN pip install theano jupyter
|
|
|
|
# And the startup script
|
|
COPY . /
|
|
|
|
# Set up Theano for the GPU
|
|
ENV THEANO_FLAGS='floatX=float32,device=gpu'
|
|
|
|
EXPOSE 8888
|