Commit 67a20245 authored by muyangli's avatar muyangli
Browse files

docs: update torch versions

parent 2bfd3ad9
...@@ -130,10 +130,10 @@ If you're using a Blackwell GPU (e.g., 50-series GPUs), install a wheel with PyT ...@@ -130,10 +130,10 @@ If you're using a Blackwell GPU (e.g., 50-series GPUs), install a wheel with PyT
pip install peft opencv-python gradio spaces GPUtil pip install peft opencv-python gradio spaces GPUtil
``` ```
To enable NVFP4 on Blackwell GPUs (e.g., 50-series GPUs), please install nightly PyTorch with CUDA 12.8. The installation command can be: To enable NVFP4 on Blackwell GPUs (e.g., 50-series GPUs), please install nightly PyTorch>=2.7 with CUDA>=12.8. The installation command can be:
```shell ```shell
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
``` ```
2. Install `nunchaku` package: 2. Install `nunchaku` package:
......
...@@ -126,10 +126,10 @@ pip install https://huggingface.co/mit-han-lab/nunchaku/resolve/main/nunchaku-0. ...@@ -126,10 +126,10 @@ pip install https://huggingface.co/mit-han-lab/nunchaku/resolve/main/nunchaku-0.
pip install peft opencv-python gradio spaces GPUtil pip install peft opencv-python gradio spaces GPUtil
``` ```
Blackwell用户需安装PyTorch nightly(CUDA 12.8 Blackwell用户需安装PyTorch>=2.7, CUDA>=12.8:
```shell ```shell
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
``` ```
2. 编译安装: 2. 编译安装:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment