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

install pybind11 through pip to get version 2.6 (#173)

pull/175/head
Yi Zhang 2021-02-28 19:19:03 -05:00 committed by GitHub
parent 79a3f639bf
commit 2c2286034b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docker/pytorch-nightly/Dockerfile
View File

@ -7,7 +7,7 @@ RUN apt update
# Python and build system.
RUN apt install -y python3 python-is-python3
RUN apt install -y python3-pip pybind11-dev python3-pybind11
RUN apt install -y python3-pip
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
@ -15,6 +15,7 @@ RUN ln -s /usr/bin/llvm-symbolizer-10 /usr/bin/llvm-symbolizer
# Install PyTorch
# Installs under: /usr/local/lib/python3.8/dist-packages/torch
RUN pip3 install numpy
RUN pip3 install pybind11
RUN pip3 install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
RUN ln -s /usr/local/lib/python3.8/dist-packages/torch /pytorch