Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -14,6 +14,10 @@ import torch
|
|
14 |
import pkg_resources
|
15 |
print([p.project_name for p in pkg_resources.working_set])
|
16 |
|
|
|
|
|
|
|
|
|
17 |
cuda_available = torch.cuda.is_available()
|
18 |
print(f"Is CUDA available: {cuda_available}")
|
19 |
if cuda_available:
|
|
|
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}")
|
23 |
if cuda_available:
|