in linux/cuda_installer/os_installers/__init__.py [0:0]
def _install_cuda_binary(self):
logger.info("Downloading CUDA Toolkit package...")
installer_path = self.download_cuda_toolkit_installer()
logger.info("Installing CUDA toolkit...")
self.run(f"sh {installer_path} --silent --toolkit", check=True)
logger.info("CUDA toolkit installation completed!")