Update torch install

docs/install-cpu-pip.md

@@ -4,7 +4,7 @@ pip install -U setuptools wheel
 
 # CPU version of pytorch has smaller footprint - see installation instructions in
 # pytorch documentation - https://pytorch.org/get-started/locally/
-pip install torchvision~=0.15.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cpu
+pip install torch==2.0.1+cpu torchvision==0.15.2+cpu --index-url https://download.pytorch.org/whl/cpu
 
 pip install autogluon
 ```

docs/install-cpu-source.md

@@ -4,7 +4,7 @@ pip install -U setuptools wheel
 
 # CPU version of pytorch has smaller footprint - see installation instructions in
 # pytorch documentation - https://pytorch.org/get-started/locally/
-pip install torchvision~=0.15.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cpu
+pip install torch==2.0.1+cpu torchvision==0.15.2+cpu --index-url https://download.pytorch.org/whl/cpu
 
 git clone https://github.com/autogluon/autogluon
 cd autogluon && ./full_install.sh

docs/install-gpu-pip.md

@@ -3,7 +3,7 @@ pip install -U pip
 pip install -U setuptools wheel
 
 # Install the proper version of PyTorch following https://pytorch.org/get-started/locally/
-pip install torchvision~=0.15.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu118
+pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118
 
 pip install autogluon
 ```

docs/install-gpu-source.md

@@ -3,7 +3,7 @@ pip install -U pip
 pip install -U setuptools wheel
 
 # Install the proper version of PyTorch following https://pytorch.org/get-started/locally/
-pip install torchvision~=0.15.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu118
+pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118
 
 git clone https://github.com/autogluon/autogluon
 cd autogluon && ./full_install.sh