Spaces:
Running
Running
David Krajewski
commited on
Commit
•
ae2d0a0
1
Parent(s):
2a0f476
Try just cudnn?
Browse files- app.py +2 -2
- packages.txt +0 -1
app.py
CHANGED
|
@@ -648,9 +648,9 @@ class Drag:
|
|
| 648 |
|
| 649 |
with gr.Blocks() as demo:
|
| 650 |
import os
|
| 651 |
-
import nvidia.cublas.lib
|
| 652 |
import nvidia.cudnn.lib
|
| 653 |
-
print(os.path.dirname(nvidia.cublas.lib.__file__))
|
| 654 |
print(os.path.dirname(nvidia.cudnn.lib.__file__))
|
| 655 |
|
| 656 |
|
|
|
|
| 648 |
|
| 649 |
with gr.Blocks() as demo:
|
| 650 |
import os
|
| 651 |
+
# import nvidia.cublas.lib
|
| 652 |
import nvidia.cudnn.lib
|
| 653 |
+
# print(os.path.dirname(nvidia.cublas.lib.__file__))
|
| 654 |
print(os.path.dirname(nvidia.cudnn.lib.__file__))
|
| 655 |
|
| 656 |
|
packages.txt
CHANGED
|
@@ -1 +0,0 @@
|
|
| 1 |
-
libcublas11
|
|
|
|
|
|