Package
 default
nccl2-2.1.4--cuda-9.0.176
Module
Version
Flavor
cuda-9.0.176
Previous versions
Next versions
Other branches
Other flavors
Role
 default
A pre-configured and fully integrated minimal runtime environment with PyTorch, an open source machine learning library for Python, Jupyter Notebook, a browser-based interactive notebook for programming, mathematics, and data science, and the Python programming language. The stack is optimized for running on NVidia GPU.
pytorch:0.3.1, python:3.6.3, jupyter_notebook:1.0.0, cuda:9.0.176, cudnn:7.0.5, cuda_only-nvidia_drivers:384.111, development_preset:1
 default
A pre-configured and fully integrated minimal runtime environment with PyTorch, an open source machine learning library for Python, Jupyter Notebook, a browser-based interactive notebook for programming, mathematics, and data science, and the Python programming language. The stack is optimized for running on NVidia GPU.
pytorch:0.3.1, python:2.7.14, jupyter_notebook:1.0.0, cuda:9.0.176, cudnn:7.0.5, cuda_only-nvidia_drivers:384.111, development_preset:1
Machine learning
A pre-configured and fully integrated software stack with PyTorch, an open source machine learning library, and Python 3.6. It provides a stable and tested execution environment for training, inference, or running as an API service. The stack can be easily integrated into continuous integration and deployment workflows. It is designed for short and long-running high-performance tasks and optimized for running on NVidia GPU.
pytorch:0.3.0, python:3.6.3, cuda:9.0.176, cudnn:7.0.5, cuda_only-nvidia_drivers:384.111, selfmanagement_preset, development_preset:1