This repository has been archived by the owner on Apr 28, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 211
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
PyTorch Integration with Tensor Comprehensions
- Loading branch information
Showing
93 changed files
with
5,419 additions
and
189 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,68 @@ | ||
FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu14.04 | ||
|
||
ENV DEBIAN_FRONTEND noninteractive | ||
|
||
RUN apt-get update | ||
|
||
RUN apt-get install -y --no-install-recommends make git ssh realpath wget unzip cmake3 vim | ||
RUN apt-get install -y --no-install-recommends libgoogle-glog-dev libyaml-dev | ||
RUN apt-get install -y --no-install-recommends libgtest-dev libz-dev libgmp3-dev | ||
RUN apt-get install -y --no-install-recommends automake libtool valgrind subversion | ||
RUN apt-get install -y --no-install-recommends ca-certificates software-properties-common | ||
|
||
RUN cmake --version | ||
|
||
# GCC 4.8.* | ||
RUN add-apt-repository ppa:ubuntu-toolchain-r/test | ||
RUN apt-get update | ||
RUN apt-get install -y --no-install-recommends libcilkrts5 gcc-4.8 g++-4.8 | ||
RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 50 | ||
RUN update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 50 | ||
|
||
ENV CC /usr/bin/gcc | ||
ENV CXX /usr/bin/g++ | ||
|
||
# LLVM+Clang | ||
ENV LLVM_SOURCES /tmp/llvm_sources-tapir5.0 | ||
WORKDIR $LLVM_SOURCES | ||
|
||
ENV CLANG_PREFIX /usr/local/clang+llvm-tapir5.0 | ||
ENV CMAKE_VERSION cmake | ||
|
||
RUN git clone --recursive https://github.com/wsmoses/Tapir-LLVM llvm && \ | ||
mkdir -p ${LLVM_SOURCES}/llvm_build && cd ${LLVM_SOURCES}/llvm_build && \ | ||
${CMAKE_VERSION} -DLLVM_ENABLE_EH=ON -DLLVM_ENABLE_OCAMLDOC=OFF -DLLVM_INSTALL_OCAMLDOC_HTML_DIR=/tmp -DLLVM_OCAML_INSTALL_PATH=/tmp -DCMAKE_INSTALL_PREFIX=${CLANG_PREFIX} -DLLVM_TARGETS_TO_BUILD=X86 -DCOMPILER_RT_BUILD_CILKTOOLS=OFF -DLLVM_ENABLE_CXX1Y=ON -DLLVM_ENABLE_TERMINFO=OFF -DLLVM_BUILD_TESTS=OFF -DLLVM_ENABLE_ASSERTIONS=ON -DCMAKE_BUILD_TYPE=Release -DLLVM_BUILD_LLVM_DYLIB=ON -DLLVM_ENABLE_RTTI=ON ../llvm/ && \ | ||
make -j"$(nproc)" -s && make install -j"$(nproc)" -s | ||
|
||
RUN rm -Rf ${LLVM_SOURCES} | ||
|
||
# Anaconda3 | ||
ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:/usr/local/cuda/targets/x86_64-linux/lib/stubs/:$LD_LIBRARY_PATH | ||
ENV PATH /usr/local/bin:/usr/local/cuda/bin:$PATH | ||
|
||
WORKDIR /conda-install | ||
RUN echo 'export PATH=/opt/conda/bin:$PATH' > /etc/profile.d/conda.sh &&\ | ||
wget --quiet https://repo.continuum.io/archive/Anaconda3-5.0.1-Linux-x86_64.sh -O anaconda.sh && \ | ||
chmod +x anaconda.sh && \ | ||
./anaconda.sh -b -p /opt/conda && \ | ||
rm anaconda.sh | ||
|
||
ENV PATH /opt/conda/bin:$PATH | ||
|
||
RUN conda install numpy decorator six future cmake pyyaml | ||
|
||
# Protobuf 3.4* | ||
WORKDIR /proto-install | ||
RUN wget --quiet https://github.com/google/protobuf/archive/v3.4.0.zip -O proto.zip && unzip -qq proto.zip -d / | ||
|
||
RUN cd /protobuf-3.4.0 && ./autogen.sh && ./configure && make -j 8 | ||
RUN cd /protobuf-3.4.0 && make install && ldconfig | ||
|
||
RUN which conda | ||
RUN conda --version | ||
RUN which protoc | ||
RUN protoc --version | ||
RUN which python | ||
RUN python --version | ||
|
||
CMD ["bash"] |
Oops, something went wrong.