Module
Pytorch
pytorch_preset
Branches
Machine learning
A pre-configured and fully integrated software stack with PyTorch, an open source software library for machine learning, and the Python programming language. The stack is designed for short and long-running high-performance tasks and is optimized for running on NVidia GPU.
pytorch:0.4.0, python:3.6.3, cuda:9.1.85.3, cudnn:7.1.3, cuda_only-nvidia_drivers:396.24, development_preset:1
Machine learning
A pre-configured and fully integrated software stack with PyTorch, an open source software library for machine learning, and the Python programming language. The stack is designed for short and long-running high-performance tasks and is optimized for running on NVidia GPU.
pytorch:0.4.0, python:2.7.14, cuda:9.1.85.3, cudnn:7.1.3, cuda_only-nvidia_drivers:396.24, development_preset:1
Machine learning
A pre-configured and fully integrated software stack with PyTorch, an open source software library for machine learning, and the Python programming language. The stack is designed for short and long-running high-performance tasks and is optimized for running on NVidia GPU.
pytorch:0.4.0, python:3.6.3, 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 software library for machine learning, and the Python programming language. The stack is designed for short and long-running high-performance tasks and is optimized for running on NVidia GPU.
pytorch:0.4.0, python:2.7.14, 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 software library for machine learning, and the Python programming language. The stack is designed for short and long-running high-performance tasks and is optimized for running on CPU.
pytorch:0.4.0, python:3.6.3, development_preset:1