Force upgrade pip when installing Torch

This commit is contained in:
Disty0
2025-06-10 13:59:32 +03:00
parent a6b58efe45
commit 7ccd94ed4f
2 changed files with 2 additions and 0 deletions
+1
View File
@@ -14,6 +14,7 @@
- **Torch**
- Set default to `torch==2.7.1`
- Support ROCm 6.4 with `---use-nightly`
- Force upgrade pip when installing Torch
- **SDNQ Quantization**
- Add group size support for convolutional layers
+1
View File
@@ -869,6 +869,7 @@ def check_torch():
if 'torch' in torch_command and not args.version:
if not installed('torch'):
log.info(f'Torch: download and install in progress... cmd="{torch_command}"')
install('--upgrade pip', 'pip', reinstall=True) # pytorch rocm is too large for older pip
install(torch_command, 'torch torchvision', quiet=True)
else:
try: