Skip to content

Commit

Permalink
Update torch version and validation output
Browse files Browse the repository at this point in the history
  • Loading branch information
bmaltais committed Oct 26, 2024
1 parent 0d2a976 commit 67f5f7b
Show file tree
Hide file tree
Showing 6 changed files with 20 additions and 15 deletions.
6 changes: 3 additions & 3 deletions requirements_linux.txt
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
# Custom index URL for specific packages
--extra-index-url https://download.pytorch.org/whl/cu124

torch==2.4.1+cu124
torchvision==0.19.1+cu124
xformers==0.0.28.post1
torch==2.5.0+cu124
torchvision==0.20.0+cu124
xformers==0.0.28.post2

bitsandbytes==0.44.0
tensorboard==2.15.2
Expand Down
4 changes: 2 additions & 2 deletions requirements_linux_rocm.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Custom index URL for specific packages
--extra-index-url https://download.pytorch.org/whl/rocm6.1
torch==2.4.0+rocm6.1
torchvision==0.19.0+rocm6.1
torch==2.5.0+rocm6.1
torchvision==0.20.0+rocm6.1

tensorboard==2.14.1
tensorflow-rocm==2.14.0.600
Expand Down
6 changes: 3 additions & 3 deletions requirements_pytorch_windows.txt
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
# Custom index URL for specific packages
--extra-index-url https://download.pytorch.org/whl/cu124

torch==2.4.1+cu124
torchvision==0.19.1+cu124
xformers==0.0.28.post1
torch==2.5.0+cu124
torchvision==0.20.0+cu124
xformers==0.0.28.post2

-r requirements_windows.txt
6 changes: 3 additions & 3 deletions requirements_runpod.txt
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.1+cu124
torchvision==0.19.1+cu124
torch==2.5.0+cu124
torchvision==0.20.0+cu124
xformers==0.0.28.post2

bitsandbytes==0.44.0
tensorboard==2.14.1
tensorflow==2.14.0 wheel
tensorrt
onnxruntime-gpu==1.17.1
xformers==0.0.28.post1

-r requirements.txt
11 changes: 8 additions & 3 deletions setup/setup_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,12 +161,17 @@ def install_requirements_inbulk(
log.info(f"Installing/Validating requirements from {requirements_file}...")

optional_parm += " -U" if upgrade else ""
optional_parm += " --quiet" if not show_stdout else ""

cmd = f"pip install -r {requirements_file} {optional_parm}"
if not show_stdout:
cmd += " --quiet"
cmd = [sys.executable, "-m", "pip", "install", "-r", requirements_file] + optional_parm.split()

if sys.platform.startswith("win32"):
cmd += ["|", "findstr", "/V", "Requirement already satisfied"]
else:
cmd += ["|", "grep", "-v", "Requirement already satisfied"]

run_cmd(cmd)

log.info(f"Requirements from {requirements_file} installed/validated.")


Expand Down
2 changes: 1 addition & 1 deletion setup/validate_requirements.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ def main():
requirements_file = args.requirements or "requirements_pytorch_windows.txt"
log.debug(f"Installing requirements from: {requirements_file}")
setup_common.install_requirements_inbulk(
requirements_file, show_stdout=False,
requirements_file, show_stdout=True,
# optional_parm="--index-url https://download.pytorch.org/whl/cu124"
)

Expand Down

0 comments on commit 67f5f7b

Please sign in to comment.