Update app.py
Browse files
app.py
CHANGED
|
@@ -21,6 +21,9 @@ import tempfile
|
|
| 21 |
import shutil
|
| 22 |
import tensorflow as tf
|
| 23 |
|
|
|
|
|
|
|
|
|
|
| 24 |
matplotlib.rcParams['figure.dpi'] = 400
|
| 25 |
matplotlib.rcParams['savefig.dpi'] = 400
|
| 26 |
|
|
|
|
| 21 |
import shutil
|
| 22 |
import tensorflow as tf
|
| 23 |
|
| 24 |
+
print(torch.__version__)
|
| 25 |
+
print(torch.version.cuda)
|
| 26 |
+
|
| 27 |
matplotlib.rcParams['figure.dpi'] = 400
|
| 28 |
matplotlib.rcParams['savefig.dpi'] = 400
|
| 29 |
|