Spaces:
Running
on
Zero
Running
on
Zero
macadeliccc
commited on
Commit
Β·
f4995ab
1
Parent(s):
2593c1c
test
Browse files
app.py
CHANGED
@@ -4,16 +4,12 @@ import torch
|
|
4 |
import subprocess
|
5 |
import numpy as np
|
6 |
|
7 |
-
|
8 |
-
|
9 |
-
|
|
|
10 |
@spaces.GPU
|
11 |
def start_ochat_server():
|
12 |
-
global device_available, device_count, device_name
|
13 |
-
|
14 |
-
print(device_available)
|
15 |
-
print(device_name) # <-- 'cuda:0' π€
|
16 |
-
print(device_count) # <-- 1 π€
|
17 |
|
18 |
# Command to start the ochat inference server
|
19 |
command = [
|
|
|
4 |
import subprocess
|
5 |
import numpy as np
|
6 |
|
7 |
+
print(torch.cuda.is_available()) # <-- True π€
|
8 |
+
print(torch.cuda.device_count()) # <-- 1 π€
|
9 |
+
print(torch.cuda.get_device_name()) # <-- 'A10G' π€
|
10 |
+
|
11 |
@spaces.GPU
|
12 |
def start_ochat_server():
|
|
|
|
|
|
|
|
|
|
|
13 |
|
14 |
# Command to start the ochat inference server
|
15 |
command = [
|