Conversation
|
Hi @tfoote, I’ve implemented the enhancement to the CUDA extension to skip the installation if CUDA is already present on the system. This change introduces a check using the I’ve tested the feature under the following conditions:
Looking forward to your feedback! Please let me know if there’s anything else I can improve. Best regards, |
tfoote
left a comment
There was a problem hiding this comment.
Thanks for looking at this, however this won't do what is needed. This is checking the host, not the container or the presence of cuda.
|
This needs major changes before it will be ready so I'm going to push this back to draft status. It can be rereviewed when the additional functionality has been implemented. |
This pull request enhances the CUDA extension to skip the installation step if CUDA is already present on the system. It adds a check to detect whether CUDA is installed using the nvidia-smi command, ensuring that redundant installations are avoided.
Key Changes
is_cuda_installed()to check if CUDA is already installed.get_snippet()method to: Skip generating the Dockerfile if CUDA is detected.