Spaces:
Build error
Build error
Commit
·
ad87f99
1
Parent(s):
74baaee
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,6 +13,7 @@ DROPOUT_RATE = 0.5
|
|
| 13 |
RNN_UNIT = 64
|
| 14 |
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
| 15 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
|
|
|
| 16 |
|
| 17 |
gpus = tf.config.list_physical_devices(device_type='GPU')
|
| 18 |
for gpu in gpus:
|
|
|
|
| 13 |
RNN_UNIT = 64
|
| 14 |
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
| 15 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
| 16 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
| 17 |
|
| 18 |
gpus = tf.config.list_physical_devices(device_type='GPU')
|
| 19 |
for gpu in gpus:
|