Merge pull request #5 from tenstorrent/mmanzoor/env-use-py3.11

Use python3.11 explicitly to install the requirements.
tenstorrent · Oct 30, 2024 · 958c588 · 958c588
2 parents 521520e + 13c1d05
commit 958c588
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/env/activate b/env/activate
@@ -20,7 +20,7 @@ else
     python3.11 -m venv $TTMLIR_VENV_DIR
     source $TTMLIR_VENV_DIR/bin/activate
     pip install --upgrade pip
-    pip install -r requirements.txt
+    python3.11 -m pip install -r requirements.txt
   fi
   export TTTORCH_ENV_ACTIVATED=1
   export TTMLIR_ENV_ACTIVATED=1

diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,10 @@ torch@https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.5.0%2Bcpu.cxx11.abi
 torchvision
 --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 -f https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels
+black
+ninja
+pre-commit
 pybind11
-tabulate
 pytest
+tabulate
 transformers