Update install_cuda.py

zackees · Jan 8, 2024 · b12631f · b12631f
1 parent 9001cea
commit b12631f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/install_cuda.py b/install_cuda.py
@@ -40,9 +40,7 @@
 
 # Delete the torch package if it doesn't have the cuda version
 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)
+    subprocess.run(["pip", "uninstall", "-y", "torch", "torchtext", "torchdata", "torchaudio", "torchvision", "torch-directml"], check=True)
     subprocess.run(["pip", "cache", "purge"], check=True)
 else:
     if TENSOR_CUDA_VERSION not in pip_list_stdout: