(//docker): Review comments incorporated

Anurag Dixit · Anurag Dixit · commit a1ba9d028f28 · 2021-10-19T12:57:09.000-07:00
Signed-off-by: Anurag Dixit &lt;anuragd@nvidia.com&gt;

(//py): Fix the version suffix

Signed-off-by: Anurag Dixit &lt;anuragd@nvidia.com&gt;
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE=21.08
+ARG BASE=21.06
 ARG BASE_IMG=nvcr.io/nvidia/pytorch:${BASE}-py3
 FROM ${BASE_IMG} as base
 
diff --git a/py/setup.py b/py/setup.py
@@ -234,7 +234,7 @@ def run(self):
       long_description=long_description,
       ext_modules=ext_modules,
       install_requires=[
-          'torch>=1.9.0+cu111,<1.11.0',
+          'torch>=1.9.0<1.11.0',
       ],
       setup_requires=[],
       cmdclass={

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