Fix CUDA path

utils/cuda_toolkit.py

@@ -14,7 +14,7 @@ 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-{CUDA_VERSION}'
+    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)