GPUを使わないディープラーニングSLIDE

qiitaに書くような記事なんですが、とりあえずこちらで。

今月爆誕しました、SLIDEは、
GPUを使わずに割と速くディープラーニングを実行するものです。
https://github.com/keroro824/HashingDeepLearning

こういうのを待っていた方は多いのではないでしょうか。
ディープラーニングをやろうとすると、
お高いGPUを買ったり、
GPUが乗ったaws等のクラウドリソースを借りたりしなければいけなかったのですが、
SLIDEはintel CPUでの高速な実行を目指したものです。
ペーパーはこちら
https://arxiv.org/abs/1903.03129


で、GPUを使うものと違い、
intelに最適化されたtensorflow等の環境構築がいりますので、
Dockerfileを作りました。以下を使ってください...

--------------------------------
FROM ubuntu:16.04
RUN apt-get update && apt-get install -y --no-install-recommends \
         build-essential \
         cmake \
         git \
         curl \
         ca-certificates \
         libjpeg-dev \
python3 \
vim \
software-properties-common \
unzip \
         libpng-dev && \
     add-apt-repository -y ppa:jonathonf/gcc-7.1 && \
     apt-get update && apt-get install -y gcc-7 g++-7 && \
     rm -rf /var/lib/apt/lists/*

WORKDIR $HOME/build
RUN git clone https://github.com/rogersce/cnpy.git .
RUN cmake .
RUN make
RUN make install

WORKDIR /workspace
RUN chmod -R a+w .

RUN curl "https://bootstrap.pypa.io/get-pip.py" -o "get-pip.py"
RUN python3 get-pip.py

RUN pip install --ignore-installed --upgrade https://github.com/lakshayg/tensorflow-build-archived/releases/download/tf-2.1.0-py35-ubuntu1604/tensorflow-2.1.0-cp35-cp35m-linux_x86_64.whl --user

RUN git clone https://github.com/keroro824/HashingDeepLearning.git
WORKDIR /workspace/HashingDeepLearning/SLIDE
RUN make

WORKDIR /workspace/HashingDeepLearning/dataset
RUN ID=0B3lPMIHmG6vGdUJwRzltS1dvUVk && \
    curl -sc /tmp/cookie "https://drive.google.com/uc?id=${ID}&export=download" > /dev/null && \
    CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)" && \
    curl -Lb /tmp/cookie "https://drive.google.com/uc?id=${ID}&export=download&confirm=${CODE}" -o Amazon.zip && \
    unzip Amazon.zip

----------------------------
enjoy!

Comments

Popular posts from this blog

conda clean --allはやってはいけない

MacbookやiMacのSSD/HDDを完全消去する(High Sierra対応版)

意外とめんどいmac OSXでのddコマンド