From ff53cac99dc71a66b1ada43d63d1307debd592a7 Mon Sep 17 00:00:00 2001 From: tysam-code <120343106+tysam-code@users.noreply.github.com> Date: Wed, 16 Aug 2023 16:33:13 -0400 Subject: [PATCH] Update README.md Removed `python -m pip install -r requirements.txt` from the default given run command to avoid borking people's pre-existing torch and torchvision installations (more likely+critical than someone not having an installation). Added some documentation for people new to torch/torchvision. --- README.md | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index fbf63aa..78f0b65 100644 --- a/README.md +++ b/README.md @@ -6,8 +6,13 @@ Welcome to the hyperlightspeedbench CIFAR-10 (HLB-CIFAR10) repo. ### How to Run -`git clone https://github.com/tysam-code/hlb-CIFAR10 && cd hlb-CIFAR10 && python -m pip install -r requirements.txt && python main.py` +`git clone https://github.com/tysam-code/hlb-CIFAR10 && cd hlb-CIFAR10 && python main.py` +Requires CUDA. Can be modified for other platforms with mild difficulty. + +If you don't have torch and/or torchvision installed, please run: + +`python -m pip install -r requirements.txt` after cd-ing into the hlb-CIFAR10 folder, then `python main.py` instead. If you're curious, this code is generally Colab friendly (in fact -- most of this was developed in Colab!). Just be sure to uncomment the reset block at the top of the code.