Not needed anymore

app.py

@@ -1,15 +1,12 @@
 """
 """
 from datetime import datetime
+t0 = datetime.now()
 
 # Upgrade PyTorch
 import os
 os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" torchvision spaces')
-
-# CUDA toolkit install
-t0 = datetime.now()
-from utils.cuda_toolkit import install_cuda_toolkit; install_cuda_toolkit()
-print('install_cuda_toolkit', -(t0 - (t0 := datetime.now())))
+print('torch upgrade', -(t0 - (t0 := datetime.now())))
 
 # Actual app.py
 import os
@@ -19,7 +16,7 @@ import spaces
 import torch
 from diffusers import FluxPipeline
 
-from utils.zerogpu import aoti_compile
+from zerogpu import aoti_compile
 
 
 pipeline = FluxPipeline.from_pretrained('black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cuda')

utils/cuda_toolkit.py

@@ -1,32 +0,0 @@
-"""
-"""
-
-import os
-import subprocess
-from pathlib import Path
-
-
-CUDA_VERSION = '12.4.0'
-DRIVERS_VERSION = '550.54.14'
-
-
-def install_cuda_toolkit():
-    installer_name = f'cuda_{CUDA_VERSION}_{DRIVERS_VERSION}_linux.run'
-    installer_url = f'https://developer.download.nvidia.com/compute/cuda/{CUDA_VERSION}/local_installers/{installer_name}'
-    installer_path = f'/tmp/{installer_name}'
-    cuda_path = f'/usr/local/cuda-12.4'
-    bashrc_path = Path.home() / '.bashrc'
-
-    subprocess.run(f'wget --progress=dot:giga {installer_url} -O {installer_path}', check=True, shell=True)
-    subprocess.run(f'fakeroot sh {installer_path} --silent --toolkit --override', check=True, shell=True)
-
-    bashrc_lines = (
-        f'# CUDA toolkit',
-        f'export PATH={cuda_path}/bin:$PATH',
-        f'export LD_LIBRARY_PATH={cuda_path}/lib64:$LD_LIBRARY_PATH'
-    )
-    with open(bashrc_path, 'a') as f:
-        f.write('\n' + '\n'.join(bashrc_lines) + '\n')
-
-    os.environ['PATH'] = f"{cuda_path}/bin:{os.environ.get('PATH', '')}"
-    os.environ['LD_LIBRARY_PATH'] = f"{cuda_path}/lib64:{os.environ.get('LD_LIBRARY_PATH', '')}"