Skip to content

Commit

Permalink
Fixed version check
Browse files Browse the repository at this point in the history
  • Loading branch information
Masahiro Tanaka committed Oct 15, 2022
1 parent e954d98 commit ff3d8aa
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 9 deletions.
14 changes: 7 additions & 7 deletions documents/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -16,27 +16,27 @@ RaNNC requires the following libraries and tools at runtime.
Installation
------------

The current version (``0.7.4``) of RaNNC requires PyTorch v1.11.0.
The current version (``0.7.5``) of RaNNC requires PyTorch v1.11.0.
``pip`` packages for ``linux_x86_64`` are available for the following combinations of Python and CUDA versions.

* Python version: 3,7, 3.8, 3.9, 3.10
* CUDA version: 11.3

The following commands install PyToch and RaNNC with pip.
(The package version should be specified as ``0.7.4+cu[CUDA_VERSION_WITHOUT_DOT]``)
(The package version should be specified as ``0.7.5+cu[CUDA_VERSION_WITHOUT_DOT]``)

.. code-block:: bash
pip install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
pip install pyrannc==0.7.4+cu113 -f https://nict-wisdom.github.io/rannc/installation.html
pip install pyrannc==0.7.5+cu113 -f https://nict-wisdom.github.io/rannc/installation.html
Use the following links to manually download the packages.

* :download:`For Python 3.7 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.4/pyrannc-0.7.4+cu113-cp37-cp37m-linux_x86_64.whl>`
* :download:`For Python 3.8 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.4/pyrannc-0.7.4+cu113-cp38-cp38-linux_x86_64.whl>`
* :download:`For Python 3.9 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.4/pyrannc-0.7.4+cu113-cp39-cp39-linux_x86_64.whl>`
* :download:`For Python 3.10 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.4/pyrannc-0.7.4+cu113-cp310-cp310-linux_x86_64.whl>`
* :download:`For Python 3.7 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.5/pyrannc-0.7.5+cu113-cp37-cp37m-linux_x86_64.whl>`
* :download:`For Python 3.8 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.5/pyrannc-0.7.5+cu113-cp38-cp38-linux_x86_64.whl>`
* :download:`For Python 3.9 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.5/pyrannc-0.7.5+cu113-cp39-cp39-linux_x86_64.whl>`
* :download:`For Python 3.10 / CUDA 11.3 <https://github.com/nict-wisdom/rannc/releases/download/v0.7.5/pyrannc-0.7.5+cu113-cp310-cp310-linux_x86_64.whl>`

If the above packages do not match your Python/CUDA versions, create a suitable package using ``Makefile``
in ``docker/``. ``make.sh`` shows the commands to create wheel packages.
2 changes: 1 addition & 1 deletion pyrannc/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

torch_ver = re.sub(r"\+.*", "", torch.__version__)
build_torch_ver = re.sub(r"\+.*", "", BUILD_TORCH_VER)
if torch_version != build_torch_ver:
if torch_ver != build_torch_ver:
print("RaNNC was compiled with PyTorch {}, but the current PyTorch version is {}.".format(
BUILD_TORCH_VER, torch.__version__), file=sys.stderr)
raise e
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def build_extension(self, ext):
subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp)


VERSION = "0.7.4"
VERSION = "0.7.5"

if "CUDA_VERSION" in os.environ:
version_nums = os.environ["CUDA_VERSION"].split(".")
Expand Down

0 comments on commit ff3d8aa

Please # to comment.