Skip to content

Commit

Permalink
Add cupy installation guide for developer installation instructions (p…
Browse files Browse the repository at this point in the history
  • Loading branch information
henryw7 authored Aug 30, 2024
1 parent 876df27 commit a065d7d
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,13 @@ cmake --build build/temp.gpu4pyscf -j 4
CURRENT_PATH=`pwd`
export PYTHONPATH="${PYTHONPATH}:${CURRENT_PATH}"
```
Then install cutensor for acceleration
Then install cutensor and cupy for acceleration (please switch the versions according to your nvcc version!)
```sh
pip3 install cutensor-cu11
pip3 install cutensor-cu12 cupy-cuda12x
```
There shouldn't be cupy or cutensor compilation during pip install process. If you see the following warning at the beginning of a gpu4pyscf job, it implies problems with cupy and cutensor installation (likely a version mismatch, or multiple versions of same package installed).
```
<repo_path>/gpu4pyscf/lib/cutensor.py:<line_number>: UserWarning: using cupy as the tensor contraction engine.
```

The package also provides multiple dockerfiles in ```dockerfiles```. One can use them as references to create the compilation envrionment.
Expand Down

0 comments on commit a065d7d

Please sign in to comment.