diff --git a/Dockerfile b/Dockerfile index 544af663fc..c4a67059f3 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,7 +7,6 @@ RUN NVTE_FRAMEWORK=pytorch CMAKE_BUILD_PARALLEL_LEVEL=4 MAX_JOBS=4 pip install g ARG BRANCH_NAME ARG DEP_GROUPS -ARG TE_COMMIT ARG KEEP_FOUNDRY=false ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0"