Spaces:
Running
on
Zero
Running
on
Zero
Try fix cudnn
Browse files- app.py +1 -0
- fix_cudnn.py +10 -0
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
# Import alias module before outetts to setup whisper redirection
|
| 3 |
import alias as _alias
|
|
|
|
| 1 |
+
import fix_cudnn
|
| 2 |
import gradio as gr
|
| 3 |
# Import alias module before outetts to setup whisper redirection
|
| 4 |
import alias as _alias
|
fix_cudnn.py
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
|
| 3 |
+
os.environ["LD_LIBRARY_PATH"] = ":".join([
|
| 4 |
+
os.environ.get("LD_LIBRARY_PATH", ""),
|
| 5 |
+
"/usr/local/lib/python3.10/site-packages/nvidia/cublas/lib",
|
| 6 |
+
"/usr/local/lib/python3.10/site-packages/nvidia/cuda_cupti/lib",
|
| 7 |
+
"/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib",
|
| 8 |
+
"/usr/local/lib/python3.10/site-packages/nvidia/cuda_runtime/lib",
|
| 9 |
+
"/usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib",
|
| 10 |
+
])
|