Spaces:
Running
on
T4
Running
on
T4
Commit
•
7e2da24
1
Parent(s):
4590717
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,6 +70,10 @@ def clear_mem():
|
|
| 70 |
|
| 71 |
|
| 72 |
print("Is cuda available",torch.cuda.is_available())
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
|
| 74 |
def setup_af(seq, model_name="model_5_ptm"):
|
| 75 |
clear_mem()
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
print("Is cuda available",torch.cuda.is_available())
|
| 73 |
+
stream = os.popen('nvcc --version')
|
| 74 |
+
output = stream.read()
|
| 75 |
+
print(output)
|
| 76 |
+
|
| 77 |
|
| 78 |
def setup_af(seq, model_name="model_5_ptm"):
|
| 79 |
clear_mem()
|