Deep learning and AI frameworks for the Azure Data Science VM

Deep learning frameworks on the DSVM are listed below.

CUDA, cuDNN, NVIDIA Driver

Category Value
Version(s) supported 11
Supported DSVM editions Windows Server 2019
Ubuntu 18.04
How is it configured / installed on the DSVM? nvidia-smi is available on the system path.
How to run it Open a command prompt (on Windows) or a terminal (on Linux), and then run nvidia-smi.

Horovod

Category Value
Version(s) supported 0.21.3
Supported DSVM editions Ubuntu 18.04
How is it configured / installed on the DSVM? Horovod is installed in Python 3.5
How to run it Activate the correct environment at the terminal, and then run Python.

NVidia System Management Interface (nvidia-smi)

Category Value
Version(s) supported
Supported DSVM editions Windows Server 2019
Ubuntu 18.04
What is it for? NVIDIA tool for querying GPU activity
How is it configured / installed on the DSVM? nvidia-smi is on the system path.
How to run it On a virtual machine with GPU's, open a command prompt (on Windows) or a terminal (on Linux), and then run nvidia-smi.

PyTorch

Category Value
Version(s) supported 1.9.0 (Ubuntu 18.04, Windows 2019)
Supported DSVM editions Windows Server 2019
Ubuntu 18.04
How is it configured / installed on the DSVM? Installed in Python, conda environments 'py38_default', 'py38_pytorch'
How to run it Terminal: Activate the correct environment, and then run Python.
* JupyterHub: Connect, and then open the PyTorch directory for samples.

TensorFlow

Category Value
Version(s) supported 2.5
Supported DSVM editions Windows Server 2019
Ubuntu 18.04
How is it configured / installed on the DSVM? Installed in Python, conda environments 'py38_default', 'py38_tensorflow'
How to run it Terminal: Activate the correct environment, and then run Python.
* Jupyter: Connect to Jupyter or JupyterHub, and then open the TensorFlow directory for samples.