Added micromamba install of pyyaml

JuliaPackaging · Feb 21, 2022 · 8f79945 · 8f79945
1 parent 7ccc225
commit 8f79945
Showing 1 changed file with 11 additions and 0 deletions.
diff --git 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 \