Skip to content

Commit

Permalink
Put mamba outside
Browse files Browse the repository at this point in the history
  • Loading branch information
utensil committed Oct 14, 2024
1 parent 375d805 commit a2de967
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 5 deletions.
6 changes: 5 additions & 1 deletion justfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,11 @@

export HOMEBREW_NO_AUTO_UPDATE := "1"
export BINSTALL_DISABLE_TELEMETRY := "true"
export LIBTORCH := justfile_directory() / "yard-rs/tch-xp/micromamba/envs/tch-rs/lib/python3.11/site-packages/torch"

export MAMBA_ROOT_PREFIX := justfile_directory() / ".." / "micromamba"
micromamba := env_var('HOME') / ".local/bin/micromamba"
mm_packages := MAMBA_ROOT_PREFIX / "envs/tch-rs/lib/python3.11/site-packages"
export LIBTORCH := mm_packages / "torch"

default:
just list
Expand Down
10 changes: 6 additions & 4 deletions yard-rs/justfile
Original file line number Diff line number Diff line change
@@ -1,7 +1,11 @@
set fallback

example_xp := justfile_directory() / "example-xp"
export LIBTORCH := justfile_directory() / "tch-xp/micromamba/envs/tch-rs/lib/python3.11/site-packages/torch"

export MAMBA_ROOT_PREFIX := justfile_directory() / ".." / ".." / "micromamba"
micromamba := env_var('HOME') / ".local/bin/micromamba"
mm_packages := MAMBA_ROOT_PREFIX / "envs/tch-rs/lib/python3.11/site-packages"
export LIBTORCH := mm_packages / "torch"

default:
just list
Expand Down Expand Up @@ -144,9 +148,7 @@ mp:

# following https://github.com/ssoudan/tch-m1

export MAMBA_ROOT_PREFIX := justfile_directory() / "tch-xp" / "micromamba"
micromamba := env_var('HOME') / ".local/bin/micromamba"
mm_packages := MAMBA_ROOT_PREFIX / "envs/tch-rs/lib/python3.11/site-packages"


[group('tch')]
prep-tch:
Expand Down

0 comments on commit a2de967

Please sign in to comment.