chore: gpu base docker image

[ThibaultFy]

Related issue

# followed by the number of the issue

Summary

Notes

Please check if the PR fulfills these requirements

• [ ] If the feature has an impact on the user experience, the changelog has been updated
• [ ] Tests for the changes have been added (for bug fixes / features)
• [ ] Docs have been added / updated (for bug fixes / features)
• [ ] The commit message follows the conventional commit specification