Skip to content

Commit

Permalink
Feat adv install 0.9.2 (#694)
Browse files Browse the repository at this point in the history
* add release content for 0.9.2

* feat(notebooks/tutorial): add notebook outputs for reference

* fix(pcm.py): add a check to not get custom_drift_model as None

* fix(travis/build): fix proper version for pytorch on mac osx wheel build

* fix(travis/build): fix torch version to match compatibility with osx version

* feat(docs): update advance install docs to match different GPU versions
  • Loading branch information
PabloCarmona authored Oct 10, 2024
1 parent 39581ec commit 31e3178
Showing 1 changed file with 21 additions and 5 deletions.
26 changes: 21 additions & 5 deletions docs/source/advanced_install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -28,13 +28,29 @@ AIHWKIT can also be installed using pip commands as shown below.

$ pip install aihwkit

- GPU::
- GPU:

$ wget https://aihwkit-gpu-demo.s3.us-east.cloud-object-storage.appdomain.cloud/aihwkit-0.8.0+cuda117-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
To install the GPU version, select the appropriate combination of Python and CUDA versions:

then::
$ pip install aihwkit-0.8.0+cuda117-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
- For Python 3.9 and CUDA 11.8::

$ wget https://aihwkit-gpu-demo.s3.us-east.cloud-object-storage.appdomain.cloud/aihwkit-0.9.2+cuda118-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
$ pip install aihwkit-0.9.2+cuda118-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

- For Python 3.9 and CUDA 12.1::

$ wget https://aihwkit-gpu-demo.s3.us-east.cloud-object-storage.appdomain.cloud/aihwkit-0.9.2+cuda121-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
$ pip install aihwkit-0.9.2+cuda121-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

- For Python 3.10 and CUDA 11.8::

$ wget https://aihwkit-gpu-demo.s3.us-east.cloud-object-storage.appdomain.cloud/aihwkit-0.9.2+cuda118-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
$ pip install aihwkit-0.9.2+cuda118-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

- For Python 3.10 and CUDA 12.1::

$ wget https://aihwkit-gpu-demo.s3.us-east.cloud-object-storage.appdomain.cloud/aihwkit-0.9.2+cuda121-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
$ pip install aihwkit-0.9.2+cuda121-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

.. note::

Expand Down

0 comments on commit 31e3178

Please sign in to comment.