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. |
Povratne informacije
Pošalјite i prikažite povratne informacije za