Update PyTorch installation command

2024-10-26 17:36:23 +00:00
parent 602bf19d93
commit 464c8a421f
1 changed files with 1 additions and 1 deletions
@@ -6,7 +6,7 @@ Change the index-url cuda version according to your system.
 ```
 conda create -n fastvideo python=3.10.12
 conda activate fastvideo
-pip3 install torch==2.5.0 torchvision==0.2.0  --index-url https://download.pytorch.org/whl/cu121
+pip3 install torch==2.5.0 torchvision  --index-url https://download.pytorch.org/whl/cu121
 pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121
 pip install ninja
 pip install flash-attn --no-build-isolation