Pytorch镜像基于 NGC pytorch 镜像构建,针对 GPU 加速进行了优化,并包含一套经过验证的库,可启用和优化 GPU 性能。该容器还包含用于加速 ETL( DALI) 、训练( cuDNN 、 NCCL )和推理( TensorRT )工作负载的软件。
评价此篇文章