Introduction to Deep Learning VM

Deep Learning VM Images is a set of virtual machine images optimized for data science and machine learning tasks. All images come with key ML frameworks and tools pre-installed. You can use them out of the box on instances with GPUs to accelerate your data processing tasks.

Deep Learning VM images are available to support many combinations of framework and processor. There are currently images supporting TensorFlow Enterprise, TensorFlow, PyTorch, and generic high-performance computing, with versions for both CPU-only and GPU-enabled workflows.

To see a list of frameworks available, see Choosing an image.

Pre-installed packages

Images are based on the Debian 11 and Ubuntu 20.04 operating systems, and these images can be configured to include the following:

  • Specific frameworks (for example, TensorFlow) and supporting packages.

  • Python 3.10 with the following packages:

    • numpy
    • scipy
    • matplotlib
    • pandas
    • nltk
    • pillow
    • scikit-image
    • opencv-python
    • scikit-learn
    • many more
  • JupyterLab notebook environments for quick prototyping

  • Nvidia packages with the latest Nvidia driver for GPU-enabled instances:

    • CUDA 11.x and 12.x (the version depends on the framework)
    • CuDNN 7.x and NCCL 2.x (the version depends on the CUDA version)

Updates

Deep Learning VM images are updated regularly with bug fixes and package updates. Check the release notes for information about updates.

Community support

Ask a question about Deep Learning VM on Stack Overflow or join the google-dl-platform Google group to discuss Deep Learning VM.

Learn more about getting support from the community.

What's next

To get started using Deep Learning VM, create a new instance using the Cloud Marketplace or using the command line.