Revert dockerfile changes

Signed-off-by: smajumdar <[email protected]>
NVIDIA · Apr 2, 2021 · 0d2c041 · 0d2c041
1 parent 4848e4c
commit 0d2c041
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:21.03-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:21.02-py3
 
 
 # build an image that includes only the nemo dependencies, ensures that dependencies
@@ -68,7 +68,7 @@ RUN COMMIT_SHA=f546575109111c455354861a0567c8aa794208a2 && \
     rm -rf ../tests test ../tensorflow_binding
 
 # uninstall stuff from base container
-RUN pip uninstall -y sacrebleu torchtext
+RUN pip uninstall -y sacrebleu
 
 # install nemo dependencies
 WORKDIR /tmp/nemo