Skip to content

Commit

Permalink
Update GPU docker file to python 3
Browse files Browse the repository at this point in the history
  • Loading branch information
wasserth committed Sep 17, 2019
1 parent e4b575a commit 255cab9
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions Dockerfile_GPU
Original file line number Diff line number Diff line change
Expand Up @@ -68,12 +68,12 @@ RUN echo '{ \
\n}' > /neurodocker/neurodocker_specs.json

RUN apt-get update \
&& apt-get -y install git g++ python python-numpy libeigen3-dev zlib1g-dev libqt4-opengl-dev libgl1-mesa-dev libfftw3-dev libtiff5-dev curl \
&& apt-get -y install git-core python-setuptools python-dev build-essential \
&& apt-get -y install git g++ python3 libeigen3-dev zlib1g-dev libqt4-opengl-dev \
&& apt-get -y install libgl1-mesa-dev libfftw3-dev libtiff5-dev curl \
&& apt-get -y install git-core python3-pip python3-dev build-essential \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*


## NVIDIA CUDA Installation
###########################
# LICENSE:
Expand Down Expand Up @@ -159,9 +159,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
rm -rf /var/lib/apt/lists/*


RUN easy_install pip \
RUN pip3 install --upgrade pip \
&& pip install wheel numpy scipy nilearn matplotlib scikit-image nibabel \
&& pip install http://download.pytorch.org/whl/cu91/torch-1.0.0-cp27-cp27mu-linux_x86_64.whl
&& pip install http://download.pytorch.org/whl/cpu/torch-1.1.0-cp35-cp35m-linux_x86_64.whl

# This command does not get cached -> very slow each time when building container -> use prebuild mrtrix_RC3.tar.gz instead
#RUN mkdir /code && cd /code \
Expand Down

0 comments on commit 255cab9

Please sign in to comment.