Fix an error when upgrading from cuda 12.4 to cuda 12.8

This commit is contained in:
oobabooga 2025-06-10 21:08:18 -07:00
parent 801db438b0
commit 13a5288d01

View file

@ -298,7 +298,7 @@ def update_pytorch_and_python():
def clean_outdated_pytorch_cuda_dependencies():
patterns = ["cu121", "cu122", "torch2.4"]
patterns = ["cu121", "cu122", "torch2.4", "torchvision", "torchaudio"]
result = run_cmd("python -m pip list --format=freeze", capture_output=True, environment=True)
matching_packages = []