Update BitBLAS README.md with additional installation instructions

microsoft · Apr 19, 2024 · 85569ee · 85569ee
1 parent 7de7990
commit 85569ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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` in the root directory. 
+  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
@@ -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. 
+**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