diff --git a/README.md b/README.md index 42a2c3e49..ef918504f 100644 --- a/README.md +++ b/README.md @@ -76,7 +76,7 @@ GPU execution requires the NVIDIA libraries cuBLAS 11.x and cuDNN 8.x to be inst ### 2. Install PyTorch, e.g. for Linux and Windows CUDA11.8: -`conda install pytorch==2.0.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia` +`conda install pytorch==2.0.0 torchaudio==2.0.0 torchvision=0.15.0 pytorch-cuda=11.8 -c pytorch -c nvidia` See other methods [here.](https://pytorch.org/get-started/previous-versions/#v200)