nivirx/mltoys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add cpu pytorch instructions

Browse Source
This commit is contained in:
Elaina Claus
2025-12-20 01:59:50 +00:00
parent 5e5ad1bc20
commit c3fbc44a34
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@@ -25,11 +25,11 @@ source .venv/bin/activate
For instructions installing pytorch refer to the [PyTorch Home page] For instructions installing pytorch refer to the [PyTorch Home page]
```bash ```bash
pip3 install numpy pip3 install numpy pandas matplotlib
# use the nvidia CUDA or CPU only packages if required. # I use the ROCm packages
# I use the ROCm packages, so the repo uses the ROCm packages.
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.4 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 ## Running the code