diff --git a/README.md b/README.md
index a1cbde8..a44d381 100644
--- a/README.md
+++ b/README.md
@@ -25,11 +25,11 @@ source .venv/bin/activate
 For instructions installing pytorch refer to the [PyTorch Home page]
 
 ```bash
-pip3 install numpy
-# use the nvidia CUDA or CPU only packages if required.
-# I use the ROCm packages, so the repo uses the ROCm packages.
+pip3 install numpy pandas matplotlib
+# I use the ROCm packages
 pip3 install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.4
-pip3 install pandas matplotlib
+# if you need the CPU only package
+# pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 ```
 
 ## Running the code