fix: torch install on linux (#615)

encord-team · Sep 4, 2023 · eaba87b · eaba87b
1 parent 8990759
commit eaba87b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -123,7 +123,7 @@ pyinstrument = "^4.5.0"
 
 [tool.poe.tasks]
 importtime = { shell = "python -X importtime -c 'import encord_active.cli.main' 2> imports.prof && python -m tuna imports.prof" }
-torch-linux = 'python -m pip install torch==2.0.1+cpu torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/torch_stable.html'
+torch-linux = 'python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu'
 
 [tool.prisma]
 use_global_node = false