feat(//notebooks): Update notebooks container for 0.1.0

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

.gitignore

@@ -36,4 +36,5 @@ dist
 bdist
 py/trtorch/_version.py
 py/wheelhouse
-py/.eggs
+py/.eggs
+notebooks/.ipynb_checkpoints/

notebooks/Dockerfile.notebook

@@ -7,8 +7,8 @@ RUN apt update && apt install bazel-3.4.1
 RUN ln -s /usr/bin/bazel-3.4.1 /usr/bin/bazel
 
 RUN pip install pillow==4.3.0
-RUN pip install torch==1.5.1
-RUN pip install torchvision==0.6.1
+RUN pip install torch==1.6.0
+RUN pip install torchvision==0.7.0
 RUN pip install notebook
 
 COPY . /workspace/TRTorch

notebooks/WORKSPACE.notebook

@@ -53,16 +53,16 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.5.1.zip"],
-    sha256 = "cf0691493d05062fe3239cf76773bae4c5124f4b039050dbdd291c652af3ab2a"
+    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.6.0.zip"],
+    sha256 = "fded948bd2dbee625cee33ebbd4843a69496729389e0200a90fbb667cdaeeb69"
 )
 
 http_archive(
     name = "libtorch_pre_cxx11_abi",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    sha256 = "818977576572eadaf62c80434a25afe44dbaa32ebda3a0919e389dcbe74f8656",
-    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.5.1.zip"],
+    sha256 = "141bb229f4bbf905541096cf8705785e7b0c79e37ca1e5db9d372730b1b9abd7",
+    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.6.0.zip"],
 )
 
 ####################################################################################