Install correct version of PyTorch based on Python version

jhc13 · Jan 15, 2024 · f87dba1 · f87dba1
1 parent 132b28e
commit f87dba1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -12,7 +12,8 @@ https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/b
 
 # PyTorch
 torch==2.1.2; platform_system != "Windows"
-https://download.pytorch.org/whl/cu121/torch-2.1.2%2Bcu121-cp311-cp311-win_amd64.whl; platform_system == "Windows"
+https://download.pytorch.org/whl/cu121/torch-2.1.2%2Bcu121-cp311-cp311-win_amd64.whl; platform_system == "Windows" and python_version == "3.11"
+https://download.pytorch.org/whl/cu121/torch-2.1.2%2Bcu121-cp310-cp310-win_amd64.whl; platform_system == "Windows" and python_version == "3.10"
 
 # CogVLM dependencies
 einops==0.7.0