Skip to content

Commit

Permalink
test deletion of cudnn folder
Browse files Browse the repository at this point in the history
  • Loading branch information
gs-olive committed Jan 20, 2024
1 parent bf81c97 commit da5aeca
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions packaging/smoke_test.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
import os
import shutil
import site

python_site_packages = site.getsitepackages()[-1]
extra_cudnn = os.path.join(python_site_packages, "nvidia", "cudnn")
print(f"Deleting directory: {extra_cudnn}")
shutil.rmtree(extra_cudnn)


import tensorrt
import torch

Expand Down

0 comments on commit da5aeca

Please sign in to comment.