Update torch version and validation output

pull/2927/head
bmaltais 2024-10-26 19:43:53 -04:00
parent 0d2a9765b4
commit 67f5f7b156
6 changed files with 20 additions and 15 deletions

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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.")

View File

@ -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"
)