From 2c4623f8dac64a6c7526279a12a6df653d11884f Mon Sep 17 00:00:00 2001
From: Mingzhi Hu <49382651+y199387@users.noreply.github.com>
Date: Thu, 9 Jun 2022 08:39:44 +0800
Subject: [PATCH] Update pytorch_train.md (#4786)

* Update pytorch_train.md

Add description about torchvision version

* Update pytorch_train.md
---
 docs/readthedocs/source/doc/Nano/QuickStart/pytorch_train.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/readthedocs/source/doc/Nano/QuickStart/pytorch_train.md b/docs/readthedocs/source/doc/Nano/QuickStart/pytorch_train.md
index 1999aecf..b9a1968e 100644
--- a/docs/readthedocs/source/doc/Nano/QuickStart/pytorch_train.md
+++ b/docs/readthedocs/source/doc/Nano/QuickStart/pytorch_train.md
@@ -51,6 +51,7 @@ Note: BigDL-Nano does not install IPEX by default. You can install IPEX using th
 
 ```bash
 python -m pip install torch_ipex==1.9.0 -f https://software.intel.com/ipex-whl-stable
+python -m pip install torchvision==0.10.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 ```
 
 #### Multi-instance Training
@@ -86,4 +87,4 @@ data_transform = transforms.Compose([
 train_set = ImageFolder(train_path, data_transform)
 train_loader = DataLoader(train_set, batch_size=batch_size, shuffle=True)
 trainer.fit(module, train_loader)
-```
\ No newline at end of file
+```