diff --git a/README.md b/README.md
index dfff06b..7e6b1c3 100644
--- a/README.md
+++ b/README.md
@@ -73,7 +73,7 @@ conda activate tabsyn
 
 Install pytorch
 ```
-pip install torch torchvision torchaudio
+pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
 ```
 
 or via conda
@@ -282,4 +282,4 @@ We appreciate your citations if you find this repository useful to your research
   journal={arXiv preprint arXiv:2310.09656},
   year={2023}
 }
-```
\ No newline at end of file
+```