diff --git a/T/Torch/Torch/build_tarballs.jl b/T/Torch/Torch/build_tarballs.jl index be656fcc427..eee94fd20fb 100644 --- a/T/Torch/Torch/build_tarballs.jl +++ b/T/Torch/Torch/build_tarballs.jl @@ -9,12 +9,22 @@ version = v"1.10.2" sources = [ GitSource("https://github.com/pytorch/pytorch.git", "71f889c7d265b9636b93ede9d651c0a9c4bee191"), GitSource("https://github.com/dhairyagandhi96/Torch.jl.git", "85bd08d39e7fba29ec4a643f60dd006ed8be8ede"), + FileSource("https://micromamba.snakepit.net/api/micromamba/linux-64/0.21.1", "c907423887b43bec4e8b24f17471262c8087b7095683f41dcef4a4e24e9a3bbd"; filename = "micromamba.tar.bz2") ] # Bash recipe for building across all platforms script = raw""" cd $WORKSPACE/srcdir +mkdir micromamba +cd micromamba +tar xfj ../micromamba.tar.bz2 +export PATH=$PATH:$WORKSPACE/srcdir/micromamba/bin +./bin/micromamba shell init -s bash -p ~/micromamba +source ~/.bashrc +micromamba activate +micromamba install -y python=3.9 pyyaml typing_extensions -c conda-forge + cd $WORKSPACE/srcdir/pytorch git submodule update --init --recursive mkdir build @@ -25,6 +35,7 @@ cmake \ -DCMAKE_TOOLCHAIN_FILE=${CMAKE_TARGET_TOOLCHAIN} \ -DBUILD_CUSTOM_PROTOBUF=OFF \ -DBUILD_PYTHON=OFF \ + -DPYTHON_EXECUTABLE=`which python3` \ -DBUILD_SHARED_LIBS=ON \ -DPROTOBUF_PROTOC_EXECUTABLE=$host_bindir/protoc \ -DCAFFE2_CUSTOM_PROTOC_EXECUTABLE=$host_bindir/protoc \