docker pull intel/intel-optimized-pytorch
Intel® Optimization for PyTorch* extends the original PyTorch* framework by creating extensions that optimize performance of deep-learning models. This container contains PyTorch* and Intel® Optimization for Pytorch*.
Documentation and Sources
LEGAL NOTICE: By accessing, downloading or using this software and any required dependent software (the “Software Package”), you agree to the terms and conditions of the software license agreements for the Software Package, which may also include notices, disclaimers, or license terms for third party software included with the Software Package. Please refer to the license file for additional details.