Updated PyTorch version in Dockerfile

tohtana · Aug 10, 2021 · 56df57a · 56df57a
1 parent a0cba67
commit 56df57a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docker/ubuntu/base/Dockerfile b/docker/ubuntu/base/Dockerfile
@@ -10,7 +10,7 @@ ENV BOOST_VERSION 1.74.0
 ENV OPENUCX_VERSION 1.9.0
 ENV OPENMPI_VERSION 4.0.5
 ENV CONDA_VERSION 4.9.2
-ENV PYTORCH_VERSION 1.8.0
+ENV PYTORCH_VERSION 1.9.0
 
 SHELL ["/bin/bash", "-c"]
 

diff --git a/setup.py b/setup.py
@@ -69,7 +69,7 @@ def build_extension(self, ext):
 setup(
     name='pyrannc',
     packages=find_packages(),
-    version='0.6.2rc1',
+    version='0.6.3',
     author='Masahiro Tanaka',
     author_email='mtnk@nict.go.jp',
     description='Deep learning framework for data/model hybrid parallelism',