(//docker): Missed changes

Signed-off-by: Anurag Dixit <[email protected]>

Author: Anurag Dixit

docker/Dockerfile.21.03

@@ -29,13 +29,25 @@ git_repository(
     shallow_since = "1570114335 -0400"
 )
 
+# External dependency for trtorch if you already have precompiled binaries.
+# This is currently used in pytorch NGC container CI testing.
+local_repository(
+    name = "trtorch",
+    path = "/opt/conda/lib/python3.8/site-packages/trtorch"
+)
+
 # CUDA should be installed on the system locally
 new_local_repository(
     name = "cuda",
     path = "/usr/local/cuda",
     build_file = "@//third_party/cuda:BUILD",
 )
 
+new_local_repository(
+    name = "cublas",
+    build_file = "@//third_party/cublas:BUILD",
+    path = "/usr",
+)
 
 ####################################################################################
 # Locally installed dependencies (use in cases of custom dependencies or aarch64)

docker/Dockerfile.21.06

@@ -234,7 +234,7 @@ def run(self):
       long_description=long_description,
       ext_modules=ext_modules,
       install_requires=[
-          'torch>=1.9.0+cu111,<1.10.0',
+          'torch>=1.9.0+cu111,<1.11.0',
       ],
       setup_requires=[],
       cmdclass={