diff --git a/README.md b/README.md index c935db675..73796371b 100644 --- a/README.md +++ b/README.md @@ -69,7 +69,7 @@ We are continuously expanding the support matrix. If you have any specific requi ## Getting Started - [Installation](https://github.com/microsoft/BitBLAS/blob/main/docs/Installation.md): - To install BitBLAS, please checkout the document [installation](https://github.com/microsoft/BitBLAS/blob/main/docs/Installation.md). Also Make sure you already have the cuda toolkit (version >= 11) installed in the system. Or you can easily install from `pip install bitblas` from PyPi. Currently we only provide whl files for CUDA 12.1 and Ubuntu >= 20.04 with Python 3.8, if you are using a different version of CUDA or OS System, you may need to build BitBLAS from source. + To install BitBLAS, please checkout the document [installation](https://github.com/microsoft/BitBLAS/blob/main/docs/Installation.md). Also Make sure you already have the cuda toolkit (version >= 11) installed in the system. Or you can easily install from `pip install bitblas` from PyPi. Currently we only provide whl files for CUDA>=12.1 and Ubuntu>=20.04 with Python>=3.8, if you are using a different version of CUDA or OS System, you may need to build BitBLAS from source. - [QuickStart](https://github.com/microsoft/BitBLAS/blob/main/docs/QuickStart.md): BitBLAS provides two Python APIs to perform mixed-precision matrix multiplication: - ```bitblas.Matmul``` implements the $W_{wdtype}A_{adtype}$ mixed-precision matrix multiplication of $C_{cdtype}[M, N] = A_{adtype}[M, K] \times W_{wdtype}[N, K]$. diff --git a/docs/Installation.md b/docs/Installation.md index 84161dacd..dad252550 100644 --- a/docs/Installation.md +++ b/docs/Installation.md @@ -10,7 +10,7 @@ The easiest way to install BitBLAS is direcly from the PyPi using pip. To install the latest version, run the following command in your terminal. -**Note**: Currently, bitblas whl is only supported on Linux systems. We recommend using Ubuntu 20.04 or later version as we build the whl files on this platform. Currently we only provide whl files for CUDA 12.1 and with Python 3.8. If you are using a different version of CUDA. you may need to build BitBLAS from source. +**Note**: Currently, bitblas whl is only supported on Linux systems. We recommend using Ubuntu 20.04 or later version as we build the whl files on this platform. Currently we only provide whl files for CUDA>=12.1 and with Python>=3.8. If you are using a different version of CUDA. you may need to build BitBLAS from source. ```bash pip install bitblas