diff --git a/README.md b/README.md index a9c522e..50b4cb6 100644 --- a/README.md +++ b/README.md @@ -56,9 +56,11 @@ We recommend using a [miniforge](https://github.com/conda-forge/miniforge) envir **1. Install prerequisite if not yet met:** -``` + +```bash pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --upgrade ``` + (Or any other CUDA versions that your GPUs/driver support)