Environment:

1. Framework: PyTorch
2. Framework version: 2.1.0
3. Horovod version: 0.28.1

Checklist:

1. Did you search issues to find if somebody asked this question before? Yes
2. If your question is about docker, did you read this doc? Yes
3. Did you check if you question is answered in the troubleshooting guide? Yes

Bug report:

I created the following Dockerfile to test out whether the latest Horovod version (0.28.1) is compatible with the latest PyTorch version (2.1.0):

# Use the official Python image as the base image
FROM python:3.10-slim-buster

# Update the system and install necessary libraries
RUN apt-get update && apt-get install -y \
    build-essential \
    cmake \
    git \
    curl \
    ca-certificates \
    libjpeg-dev \
    libpng-dev && \
    rm -rf /var/lib/apt/lists/*

# Install PyTorch
RUN pip install torch==2.1.0

# Install Horovod
RUN HOROVOD_WITH_PYTORCH=1 pip install horovod==0.28.1

And it returned various errors that I am unable to decipher. Here is the full build trace: error.txt