R1ckShi commited on
Commit
e7933a9
·
verified ·
1 Parent(s): 5eefb0b
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,10 @@
12
  # See the License for the specific language governing permissions and
13
  # limitations under the License.
14
  import os
 
15
 
16
  os.system('nvidia-smi')
 
17
 
18
  import importlib
19
  import sys
@@ -35,12 +37,10 @@ def modify_dynamic_modules_file(dynamic_modules_file):
35
  modify_dynamic_modules_file(dynamic_modules_file1)
36
  modify_dynamic_modules_file(dynamic_modules_file2)
37
 
38
- import os
39
  import sys
40
  import argparse
41
  import gradio as gr
42
  import numpy as np
43
- import torch
44
  import torchaudio
45
  import random
46
  import librosa
 
12
  # See the License for the specific language governing permissions and
13
  # limitations under the License.
14
  import os
15
+ import torch
16
 
17
  os.system('nvidia-smi')
18
+ print(torch.backends.cudnn.version())
19
 
20
  import importlib
21
  import sys
 
37
  modify_dynamic_modules_file(dynamic_modules_file1)
38
  modify_dynamic_modules_file(dynamic_modules_file2)
39
 
 
40
  import sys
41
  import argparse
42
  import gradio as gr
43
  import numpy as np
 
44
  import torchaudio
45
  import random
46
  import librosa