diff --git a/docker/torchbench-nightly.dockerfile b/docker/torchbench-nightly.dockerfile index cdea00bf2e..95ecae4c3e 100644 --- a/docker/torchbench-nightly.dockerfile +++ b/docker/torchbench-nightly.dockerfile @@ -7,9 +7,6 @@ ENV CONDA_ENV=torchbench ENV SETUP_SCRIPT=/workspace/setup_instance.sh ARG TORCHBENCH_BRANCH=${TORCHBENCH_BRANCH:-main} -# Setup dependencies -RUN sudo apt install -y libsdl2-dev libsdl2-2.0-0 - # Setup Conda env and CUDA RUN git clone -b "${TORCHBENCH_BRANCH}" --single-branch \ https://github.com/pytorch/benchmark /workspace/benchmark