update README

README.md

@@ -18,14 +18,10 @@ pip install -U joyrl
 Torch GPU install:
 
 ```bash
-# pip CPU only
-pip install torch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0
-# if network error, then GPU with mirror image
-pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu113
-# CPU only
-conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cpuonly -c pytorch
-# GPU 
-conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
+# CUDA 11.8
+pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu118
+# CUDA 12.1
+pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121
 ```
 
 ## Usage

docs/README.md

@@ -10,24 +10,18 @@
 
 ```bash
 # you need to install Anaconda first
-conda create -n joyrl python=3.8
+conda create -n joyrl python=3.10
 conda activate joyrl
 pip install -U joyrl
 ```
 
-Torch install:
-
-Pip install is recommended, but if you encounter network error, you can try conda install or pip install with mirrors.
+Torch GPU install:
 
 ```bash
-# pip CPU only
-pip install torch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0
-# if network error, then GPU with mirror image
-pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu113
-# CPU only
-conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cpuonly -c pytorch
-# GPU 
-conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
+# CUDA 11.8
+pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu118
+# CUDA 12.1
+pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121
 ```
 
 ## Usage