Skip to content

Commit

Permalink
update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
wxj6000 authored Dec 18, 2023
1 parent 4fb6db5 commit ae62341
Showing 1 changed file with 14 additions and 17 deletions.
31 changes: 14 additions & 17 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,25 +4,22 @@ Installation
--------

> [!NOTE]
> The compiled binary packages support compute capability 7.0 and later (Volta and later, such as Tesla V100, RTX 20 series and later). For older GPUs, please compile the package with the source code as follows.
> The compiled binary packages support compute capability 7.0 and later (Volta and later, such as Tesla V100, RTX 20 series and later). For older GPUs (GTX 10**, Tesla P100), please compile the package with the source code as follows.
For **CUDA 11.x**
```sh
pip3 install gpu4pyscf-cuda11x
```
and install cutensor
```sh
python -m cupyx.tools.install_library --cuda 11.x --library cutensor
```
Run ```nvidia-smi``` in your terminal to check the installed CUDA version.

For **CUDA 12.x**
```sh
pip3 install gpu4pyscf-cuda12x
```
and install cutensor
```sh
python -m cupyx.tools.install_library --cuda 12.x --library cutensor
```
Choose the proper package based on your CUDA environment.

| Platform | Command |
----------------| --------------------------------------|
| **CUDA 11.x** | ```pip3 install gpu4pyscf-cuda11x``` |
| **CUDA 12.x** | ```pip3 install gpu4pyscf-cuda12x``` |

```cuTensor``` is **highly recommended** to be installed for accelerating tensor contractions.

For **CUDA 11.x**, ```python -m cupyx.tools.install_library --cuda 11.x --library cutensor```

For **CUDA 12.x**, ```python -m cupyx.tools.install_library --cuda 12.x --library cutensor```

Compilation
--------
Expand Down

0 comments on commit ae62341

Please sign in to comment.