llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/docker/pytorch-nightly/Dockerfile

35 lines
1.2 KiB
Docker
Raw Normal View History

Add docker image/instructions for building against pytorch 1.6.
2020-09-25 08:40:25 +08:00
FROM ubuntu:20.04
MAINTAINER Stella Laurenzo <stellaraccident@gmail.com>
USER root
ENV DEBIAN_FRONTEND=noninteractive
RUN apt update
# Python and build system.
RUN apt install -y python3 python-is-python3
install pybind11 through pip to get version 2.6 (#173)
2021-03-01 08:19:03 +08:00
RUN apt install -y python3-pip
Add docker image/instructions for building against pytorch 1.6.
2020-09-25 08:40:25 +08:00
RUN apt install -y git cmake ccache ninja-build
RUN apt install -y clang lld
RUN ln -s /usr/bin/llvm-symbolizer-10 /usr/bin/llvm-symbolizer
# Install PyTorch
# Installs under: /usr/local/lib/python3.8/dist-packages/torch
More progress on PyTorch acap device capture. * Now gets far enough to capture batch_norm. * Has some issues still with in-place ops. * Can materialize constants. * Includes an upgrade to PyTorch nightly, which has important bug fixes for fallback and boxed kernel dispatch. * Fixes #78, #79, #80. * Will do more testing in a follow-up once further bugs are fixed that facilitate getting at the other features.
2020-10-16 09:28:30 +08:00
RUN pip3 install numpy
install pybind11 through pip to get version 2.6 (#173)
2021-03-01 08:19:03 +08:00
RUN pip3 install pybind11
More progress on PyTorch acap device capture. * Now gets far enough to capture batch_norm. * Has some issues still with in-place ops. * Can materialize constants. * Includes an upgrade to PyTorch nightly, which has important bug fixes for fallback and boxed kernel dispatch. * Fixes #78, #79, #80. * Will do more testing in a follow-up once further bugs are fixed that facilitate getting at the other features.
2020-10-16 09:28:30 +08:00
RUN pip3 install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
Add docker image/instructions for building against pytorch 1.6.
2020-09-25 08:40:25 +08:00
RUN ln -s /usr/local/lib/python3.8/dist-packages/torch /pytorch
# Build configuration
RUN mkdir /ccache && ln -s /usr/bin/ccache /ccache/clang && ln -s /usr/bin/ccache /ccache/clang++
Add scripts/documentation for VSCode setup with a docker dev image. * Forks a subset of my shell functions into docker_shell_funcs.sh, specifically needed to create docker images that run as yourself. * Extends the readme with the three command bootstrap to get a dev container running. * Step by step instructions for configuring VSCode for Intellisense in either npcomp or LLVM. * Changes LLVM config options to enable tests. This setup is now suitable for upstream changes as well without rebuilding.
2020-10-08 09:31:24 +08:00
RUN mkdir /build && chmod a+rw /build /ccache
Add docker image/instructions for building against pytorch 1.6.
2020-09-25 08:40:25 +08:00
ENV PATH "/ccache:${PATH}"
ENV CC clang
ENV CXX clang++
# Binary distributions of torch force CXX11 ABI 0 :(
ENV CXXFLAGS "-D_GLIBCXX_USE_CXX11_ABI=0"
ENV LDFLAGS "-fuse-ld=/usr/bin/ld.lld"
ENV CMAKE_PREFIX_PATH /pytorch/share/cmake
Add scripts/documentation for VSCode setup with a docker dev image. * Forks a subset of my shell functions into docker_shell_funcs.sh, specifically needed to create docker images that run as yourself. * Extends the readme with the three command bootstrap to get a dev container running. * Step by step instructions for configuring VSCode for Intellisense in either npcomp or LLVM. * Changes LLVM config options to enable tests. This setup is now suitable for upstream changes as well without rebuilding.
2020-10-08 09:31:24 +08:00
ENV LLVM_BUILD_DIR /build/llvm-build
ENV LLVM_INSTALL_DIR /build/llvm-install
ENV NPCOMP_BUILD_DIR /build/npcomp