meetsuki commited on
Commit
ef719fc
1 Parent(s): 51cca9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,6 +1,8 @@
1
- python3 -m pip install tensorflow
 
 
2
  # Verify install:
3
- python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
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