From c3fbc44a34d9ae0948413f4dfea5d4943cdb6e3b Mon Sep 17 00:00:00 2001
From: Elaina Claus <elaina.claus@chtm.me>
Date: Sat, 20 Dec 2025 01:59:50 +0000
Subject: [PATCH] add cpu pytorch instructions

---
 README.md | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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