Skip to content

Commit

Permalink
Make setup.py handle case where cudnn directory is created but empty
Browse files Browse the repository at this point in the history
Signed-off-by: Jonathan Swartz <[email protected]>
  • Loading branch information
swahtz committed Dec 17, 2024
1 parent 093a638 commit 39b0498
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fvdb/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,7 +236,7 @@ def download_and_install_cudnn() -> Tuple[List[str], List[str]]:
if cudnn_hash != cudnn_hash_output:
raise RuntimeError("Hash of cudnn.tar.xz does not match")

if not folder_filepath.exists():
if (not folder_filepath.exists()) or (len(os.listdir(folder_filepath)) == 0):
logging.info("Extracting cudnn…")
with tarfile.open(tar_filepath, "r:xz") as tar:
tar.extractall(folder_filepath)
Expand Down

0 comments on commit 39b0498

Please sign in to comment.