diff --git a/install_cuda.py b/install_cuda.py
index 5bf370b..b3dc25d 100644
--- a/install_cuda.py
+++ b/install_cuda.py
@@ -41,6 +41,8 @@
 # Delete the torch package if it doesn't have the cuda version
 if args.force:
     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", "cache", "purge"], check=True)
 else:
     if TENSOR_CUDA_VERSION not in pip_list_stdout: