Feature: Custom base image with CUDA

[ri-ceres]

🚀 Feature

Would like to use this for machine learning development, but the base image would need to be modified to include CUDA. Are there any plans or workarounds for this?

[hugodutka]

Hi, sorry for a delayed reply. The problem with GPUs is not only that you need drivers for them, but you also need access to the GPU itself. Firecracker has no support for virtual GPUs, so we currently can't even give Hocus access to a GPU. However, we're in the process of swapping Firecracker for QEMU, so we eventually will be able to add this feature. It's not high on our priority list though. This may change if enough users say they need it.