Merge pull request #20 from Paperspace/fix/fastai-v3

Fix/fastai v3

Author: Louis DeLosSantos

fastai-v3/Dockerfile

@@ -1,5 +1,6 @@
 FROM nvidia/cuda:9.2-base-ubuntu16.04
-
+# See http://bugs.python.org/issue19846
+ENV LANG C.UTF-8
 LABEL com.nvidia.volumes.needed="nvidia_driver"
 
 RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
@@ -18,44 +19,31 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
          libpng-dev &&\
      rm -rf /var/lib/apt/lists/*
 
+ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
 ENV PYTHON_VERSION=3.6
+
 RUN curl -o ~/miniconda.sh -O  https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh  && \
      chmod +x ~/miniconda.sh && \
      ~/miniconda.sh -b -p /opt/conda && \
      rm ~/miniconda.sh && \
     /opt/conda/bin/conda install conda-build
 
-WORKDIR /notebooks
-
-# clone fastai v1.0.42
-RUN git clone --single-branch --branch 1.0.42 https://github.com/fastai/fastai.git .
-RUN ls && /opt/conda/bin/conda env create
-
-ENV PATH /opt/conda/envs/fastai/bin:$PATH
-ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
+ENV PATH=$PATH:/opt/conda/bin/
 ENV USER fastai
+# Create Enviroment
+COPY enviroment.yaml /enviroment.yaml
+RUN conda env create -f enviroment.yaml
 
+WORKDIR /notebooks
+# Activate Source
 CMD source activate fastai
 CMD source ~/.bashrc
 
-# PyTorch 1.0 and FastAI 1.0
-RUN /opt/conda/bin/conda install --name fastai -c conda-forge jupyterlab
-RUN /opt/conda/bin/conda install --name fastai -c pytorch pytorch-nightly cuda92
-RUN /opt/conda/bin/conda install --name fastai -c fastai torchvision
-RUN /opt/conda/bin/conda install --name fastai -c fastai fastai
-RUN /opt/conda/bin/conda install --name fastai -c fastai fastprogress
-RUN /opt/conda/bin/conda clean -ya
-
 RUN chmod -R a+w /notebooks
-
-ENV PATH /opt/conda/bin:$PATH
 WORKDIR /notebooks
 
-# Clone course-v3 repo and checkout v3 master commit hash as of 1/24/2019
-RUN git clone https://github.com/fastai/course-v3.git && cd course-v3 && git checkout 2d532e82517a8b528b351fa6b2da985da0affe8b & cd ..
-RUN git checkout
-
-ENV PATH /opt/conda/envs/fastai/bin:$PATH
+# Clone course-v3
+RUN git clone https://github.com/fastai/course-v3.git
 
 COPY config.yml /root/.fastai/config.yml
 COPY run.sh /run.sh

fastai-v3/README.md

@@ -26,4 +26,4 @@ Nvidia Drivers
 
 You can also just run the following without having to build the entire container yourself. This will pull the container from Docker Hub.
 
-`sudo docker run --runtime=nvidia -d -p 8888:8888 paperspace/fastai:1.0-CUDA9.2-base-3.0`
+`sudo docker run --runtime=nvidia -d -p 8888:8888 paperspace/fastai:1.0-CUDA9.2-base-3.0-v1.0.6`

fastai-v3/enviroment.yaml

@@ -0,0 +1,15 @@
+name: fastai
+channels:
+  - conda-forge
+  - pytorch
+  - fastai
+dependencies:
+  - pytorch
+  - torchvision
+  - cuda92
+  - fastprogress
+  - fastai
+  - jupyter
+  - pip:
+      - nvidia-ml-py3
+      - dataclasses

fastai-v3/run.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
-
+source activate fastai
 ln -s /storage /notebooks/course-v3/nbs/dl1/data
 jupyter notebook --ip=0.0.0.0 --no-browser --allow-root