kiranbandi commited on
Commit
930b54d
·
verified ·
1 Parent(s): 1c0a4a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -14,9 +14,10 @@ import torch
14
  import pkg_resources
15
  print([p.project_name for p in pkg_resources.working_set])
16
 
17
- from pip._internal.utils.misc import get_installed_distributions
18
- p = get_installed_distributions()
19
- for a in p: print(a)
 
20
 
21
  cuda_available = torch.cuda.is_available()
22
  print(f"Is CUDA available: {cuda_available}")
 
14
  import pkg_resources
15
  print([p.project_name for p in pkg_resources.working_set])
16
 
17
+ import pkg_resources
18
+ dists = [str(d).replace(" ","==") for d in pkg_resources.working_set]
19
+ for i in dists:
20
+ print(i)
21
 
22
  cuda_available = torch.cuda.is_available()
23
  print(f"Is CUDA available: {cuda_available}")