Update install_cuda.py

zackees · Jan 8, 2024 · c3c92ca · c3c92ca
1 parent 838108e
commit c3c92ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/install_cuda.py b/install_cuda.py
@@ -46,7 +46,7 @@
         "torchdata",
         "torchaudio",
         "torchvision",
-        "torch-directm"
+        "torch-directml"
     ]
     for package in uninstall:
         subprocess.run(["pip", "uninstall", "-y", package], check=True)