diff --git a/docker/Dockerfile b/docker/Dockerfile index 9b604ad4fc..e66c3a9170 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -305,7 +305,6 @@ RUN if [ -n "$CUDA_VERSION" ] ; then \ cd .. && \ pip install --no-dependencies "git+ssh://git@github.com/mosaicml/megablocks-private.git@main#egg=megablocks[all]" && \ pip install --no-dependencies git+https://github.com/tgale96/grouped_gemm@v0.1.4 && \ - pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable && \ pip uninstall -y torch && \ pip uninstall -y pytorch-triton && \ pip uninstall -y torchvision && \