--- pip_packages: - apex - autogluon - autogluon-contrib-nlp - autogluon.core - autogluon.extra - autogluon.features - autogluon.mxnet - autogluon.tabular - awscli - awsio - boto3 - botocore - catboost - datasets - dgl - fastai - gluoncv - h5py - horovod - lightgbm - multi-model-server - mxnet - mxnet-model-server - numpy - opencv-python - pandas - pillow - pytorch - requests - s3fs - s3transfer - sagemaker - sagemaker-experiments - sagemaker-inference - sagemaker-mxnet-inference - sagemaker-mxnet-training - sagemaker-pytorch-inference - sagemaker-pytorch-training - sagemaker-tensorflow - sagemaker-tensorflow-inference - sagemaker-tensorflow-training - sagemaker-training - scikit-learn - scipy - seaborn - smclarify - smdebug - smdebug-rulesconfig - smdistributed-dataparallel - smdistributed-modelparallel - spacy - tensorflow - tensorflow-addons - tensorflow-gpu - tensorflow-library - tensorflow-serving-api - tensorflow-serving-api-gpu - tokenizers - torch - torchaudio - torchnet - torchserve - torch-model-archiver - torchvision - transformers - xgboost - habana-tensorflow - habana-horovod - py-cpuinfo - tensorflow-datasets - tensorflow-metadata - tf-slim - cython - imgaug - cloudpickle - munch - h5py - wrapt - bs4 - tensorflow-hub - PyYAML - boto3 - filelock - sentencepiece - packaging - lvis - gin-config - mpi4py - absl-py - protobuf - regex - cffi - typing-extensions - setuptools - ninja - mkl - mkl-include - pybind11 apt_packages: - cuda-command-line-tools - cuda-cufft - cuda-curand - cuda-cusolver - cuda-cuparse - emacs - libcublas - libcudnn - libgcc - libnccl - libnccl-dev - libstdc++ - libgcc - emacs - habanalabs-thunk - habanalabs-firmware-tools - habanalabs-graph