Don't try to load cufile (#138539)

pytorchbot · kit1980 · web-flow · commit cde6b382ff39 · 2024-10-22T10:45:53.000-07:00
Don't try to load cufile (#138501) Trying to loading it caused a big issue with 2.5.0 release - #138324 cufile is not actually used currently by default, see #133489 Pull Request resolved: #138501 Approved by: https://github.com/atalman, https://github.com/mikaylagawarecki, https://github.com/malfet (cherry picked from commit 012ff2a) Co-authored-by: Sergii Dymchenko <sdym@meta.com>
diff --git a/torch/__init__.py b/torch/__init__.py
@@ -308,7 +308,6 @@ def _load_global_deps() -> None:
             "cuda_runtime": "libcudart.so.*[0-9]",
             "cuda_cupti": "libcupti.so.*[0-9]",
             "cufft": "libcufft.so.*[0-9]",
-            "cufile": "libcufile.so.*[0-9]",
             "curand": "libcurand.so.*[0-9]",
             "nvjitlink": "libnvJitLink.so.*[0-9]",
             "cusparse": "libcusparse.so.*[0-9]",
