Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_app.py
Browse files- gradio_app.py +0 -4
gradio_app.py
CHANGED
@@ -18,10 +18,6 @@ logging.basicConfig(level=logging.DEBUG)
|
|
18 |
|
19 |
# Ensure necessary devices are available
|
20 |
print("torch.__version__: ", torch.__version__)
|
21 |
-
import subprocess
|
22 |
-
|
23 |
-
cuda_version = subprocess.check_output(['nvcc', '--version']).decode('utf-8')
|
24 |
-
print("cuda_version: ", cuda_version)
|
25 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
26 |
logger.info("device: {}".format(device))
|
27 |
|
|
|
18 |
|
19 |
# Ensure necessary devices are available
|
20 |
print("torch.__version__: ", torch.__version__)
|
|
|
|
|
|
|
|
|
21 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
22 |
logger.info("device: {}".format(device))
|
23 |
|