Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
-
|
|
|
|
|
2 |
# Verify install:
|
3 |
-
|
4 |
|
5 |
import gradio as gr
|
6 |
import tensorflow as tf
|
|
|
1 |
+
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
|
2 |
+
# Anything above 2.10 is not supported on the GPU on Windows Native
|
3 |
+
python -m pip install "tensorflow<2.11"
|
4 |
# Verify install:
|
5 |
+
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
|
6 |
|
7 |
import gradio as gr
|
8 |
import tensorflow as tf
|