Edit

Deep learning and AI frameworks for the Azure Data Science Virtual Machine

  • Article

Deep learning frameworks on the DSVM are listed here:

CUDA, cuDNN, NVIDIA Driver

Category Value
Supported versions 11
Supported DSVM editions Windows Server 2019
Linux
How is it configured and 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
Supported versions 0.21.3
Supported DSVM editions Linux
How is it configured and 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
Supported versions
Supported DSVM editions Windows Server 2019
Linux
What is it used for? As an NVIDIA tool to query GPU activity
How is it configured and 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
Supported versions 1.9.0 (Linux, Windows 2019)
Supported DSVM editions Windows Server 2019
Linux
How is it configured and installed on the DSVM? Installed in Python, conda environments 'py38_default', 'py38_pytorch'
How to run it At the terminal, activate the appropriate environment, and then run Python.
* JupyterHub: Connect, and then open the PyTorch directory for samples.

TensorFlow

Category Value
Supported versions 2.5
Supported DSVM editions Windows Server 2019
Linux
How is it configured and installed on the DSVM? Installed in Python, conda environments 'py38_default', 'py38_tensorflow'
How to run it At the terminal, activate the correct environment, and then run Python.
* Jupyter: Connect to Jupyter or JupyterHub, and then open the TensorFlow directory for samples.