yiren98 commited on
Commit
da5c9b4
·
verified ·
1 Parent(s): bbea273

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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