From 9abb1fe85f45941a80deffa4e543b8b27ccfa1f3 Mon Sep 17 00:00:00 2001
From: BVegNow <117610616+BVegNow@users.noreply.github.com>
Date: Wed, 3 Jul 2024 02:09:35 +0800
Subject: [PATCH] Update README.md

added torchvision
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 42a2c3e4..ef918504 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)