simonduerr commited on
Commit
549e82e
1 Parent(s): 042fbee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -15,7 +15,10 @@ import numpy as np
15
  import matplotlib.pyplot as plt
16
  import sys
17
  import plotly.graph_objects as go
 
18
 
 
 
19
  print(os.getcwd())
20
  if "/home/user/app/alphafold" not in sys.path:
21
  sys.path.append("/home/user/app/alphafold")
 
15
  import matplotlib.pyplot as plt
16
  import sys
17
  import plotly.graph_objects as go
18
+ import torch
19
 
20
+ print('GPU available',torch.cuda.is_available())
21
+ print('__CUDA Device Name:',torch.cuda.get_device_name(0))
22
  print(os.getcwd())
23
  if "/home/user/app/alphafold" not in sys.path:
24
  sys.path.append("/home/user/app/alphafold")