diff --git a/torch/__init__.py b/torch/__init__.py index dfd46539681..c49ae87f222 100644 --- a/torch/__init__.py +++ b/torch/__init__.py @@ -343,6 +343,7 @@ def _load_global_deps() -> None: "cufft": "libcufft.so.*[0-9]", "curand": "libcurand.so.*[0-9]", "nvjitlink": "libnvJitLink.so.*[0-9]", + "cufile": "libcufile.so.*[0-9]", "cusparse": "libcusparse.so.*[0-9]", "cusparselt": "libcusparseLt.so.*[0-9]", "cusolver": "libcusolver.so.*[0-9]",