zerchen commited on
Commit
b20a8bb
·
1 Parent(s): 462d3d7
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ LIGHT_PURPLE=(0.25098039, 0.274117647, 0.65882353)
30
  STEEL_BLUE=(0.2745098, 0.5098039, 0.7058824)
31
 
32
 
 
33
  def install_cuda_toolkit():
34
  CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
35
  CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
 
30
  STEEL_BLUE=(0.2745098, 0.5098039, 0.7058824)
31
 
32
 
33
+ @spaces.GPU()
34
  def install_cuda_toolkit():
35
  CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
36
  CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)