refactor(//docker): Address review comments

narendasan · narendasan · commit a1325971c0d3 · 2022-05-09T19:11:18.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -44,7 +44,6 @@ RUN rm -rf /opt/pytorch/torch_tensorrt
 # copy source repo
 COPY . /workspace/torch_tensorrt
 COPY --from=torch-tensorrt-builder  /workspace/torch_tensorrt/src/py/dist/ .
-#RUN conda init bash
 
 RUN pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org
 RUN jupyter nbextension enable --py widgetsnbextension
diff --git a/docker/Dockerfile.ngc b/docker/Dockerfile.ngc
@@ -1,4 +1,4 @@
-ARG BASE=21.10
+ARG BASE=22.04
 ARG BASE_IMG=nvcr.io/nvidia/pytorch:${BASE}-py3
 FROM ${BASE_IMG} as base
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-ARG BASE=21.10`
	`1`	`+ARG BASE=22.04`
`2`	`2`	`ARG BASE_IMG=nvcr.io/nvidia/pytorch:${BASE}-py3`
`3`	`3`	`FROM ${BASE_IMG} as base`
`4`	`4`