ZhengChong commited on
Commit
864fc50
1 Parent(s): 5cbfd4d

chore: Update app.py to set CUDA_HOME environment variable and add CUDA bin directory to PATH

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import argparse
2
  import os
3
  os.environ['CUDA_HOME'] = '/usr/local/cuda'
 
 
4
  from datetime import datetime
5
  import spaces
6
  import gradio as gr
 
1
  import argparse
2
  import os
3
  os.environ['CUDA_HOME'] = '/usr/local/cuda'
4
+ os.environ['PATH'] = os.environ['PATH'] + ':/usr/local/cuda/bin'
5
+
6
  from datetime import datetime
7
  import spaces
8
  import gradio as gr