install cnvidia-cuda-toolkit

app_onediff.py

@@ -43,8 +43,7 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
 #     options = json.loads('{"mode": "max-optimize:max-autotune:low-precision", "memory_format": "channels_last", "dynamic": true}')
 #     return compile_pipe(torch_module, backend="nexfort", options=options, fuse_qkv_projections=True)
 
-os.system("python3 -m pip install conda")
-os.system("conda install nvidia/label/cuda-12.1.0::cuda-toolkit")
+os.system("apt install -y nvidia-cuda-toolkit")
 print(os.environ.get('CUDA_PATH'))
 print(os.environ.get('CUDA_HOME'))
 os.system("pip show torch")