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
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
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:
......
......@@ -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
```
Blackwell用户需安装PyTorch nightly(CUDA 12.8
Blackwell用户需安装PyTorch>=2.7, CUDA>=12.8:
```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. 编译安装:
......
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