From a5851ff9db310e6f1d30c556c6b1eeefb9b4ec7c Mon Sep 17 00:00:00 2001 From: Naren Dasan Date: Mon, 7 Dec 2020 13:54:49 -0800 Subject: [PATCH] feat(//notebooks): Update notebooks container for 0.1.0 Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- .gitignore | 3 ++- notebooks/Dockerfile.notebook | 4 ++-- notebooks/WORKSPACE.notebook | 8 ++++---- 3 files changed, 8 insertions(+), 7 deletions(-) diff --git a/.gitignore b/.gitignore index 60408dd968..b75f6cb62e 100644 --- a/.gitignore +++ b/.gitignore @@ -36,4 +36,5 @@ dist bdist py/trtorch/_version.py py/wheelhouse -py/.eggs \ No newline at end of file +py/.eggs +notebooks/.ipynb_checkpoints/ \ No newline at end of file diff --git a/notebooks/Dockerfile.notebook b/notebooks/Dockerfile.notebook index 55a118fae7..ae732785c3 100644 --- a/notebooks/Dockerfile.notebook +++ b/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 diff --git a/notebooks/WORKSPACE.notebook b/notebooks/WORKSPACE.notebook index 9030af467b..25b5773226 100755 --- a/notebooks/WORKSPACE.notebook +++ b/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"], ) ####################################################################################