Skip to main content
For W&B Launch jobs that use an accelerator, provide a base image that includes the necessary accelerator components. This ensures that your job has the required GPU drivers, runtime libraries, and other hardware-specific dependencies to execute. The accelerator image must meet the following requirements:
  • Compatibility with Debian (the Launch Dockerfile uses apt-get to install Python).
  • Supported CPU and GPU hardware instruction set (confirm the CUDA version compatibility with the intended GPU).
  • Compatibility between the supplied accelerator version and the packages in the machine learning algorithm.
  • Installation of packages that require additional steps for hardware compatibility.