Update install_cuda.py

zackees · Jan 8, 2024 · 9001cea · 9001cea
1 parent 7799595
commit 9001cea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/install_cuda.py b/install_cuda.py
@@ -39,7 +39,7 @@
 ).stdout
 
 # Delete the torch package if it doesn't have the cuda version
-if args.force:
+if args.force or True:  # always do this path now.
     subprocess.run(["pip", "uninstall", "-y", "torch"], check=True)
     subprocess.run(["pip", "uninstall", "-y", "torchvision"], check=True)
     subprocess.run(["pip", "uninstall", "-y", "torch-directml"], check=True)