diff options
| author | OlivierDehaene <Olivier.dehaene@gmail.com> | 2023-10-16 16:37:38 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-16 15:37:38 +0100 |
| commit | 75629981bc2b101400a301803c027da2362a4ff9 (patch) | |
| tree | 105781868b6024facddbf05492dacd33873d4903 /candle-book | |
| parent | 0106b0b04c3505a1155b3eab65ac212977c6c3dd (diff) | |
| download | candle-75629981bc2b101400a301803c027da2362a4ff9.tar.gz candle-75629981bc2b101400a301803c027da2362a4ff9.tar.bz2 candle-75629981bc2b101400a301803c027da2362a4ff9.zip | |
feat: parse Cuda compute cap from env (#1066)
* feat: add support for multiple compute caps
* Revert to one compute cap
* fmt
* fix
Diffstat (limited to 'candle-book')
| -rw-r--r-- | candle-book/src/guide/installation.md | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-book/src/guide/installation.md b/candle-book/src/guide/installation.md index 394cef35..ca8b7968 100644 --- a/candle-book/src/guide/installation.md +++ b/candle-book/src/guide/installation.md @@ -12,6 +12,9 @@ compute_cap 8.9 ``` +You can also compile the Cuda kernels for a specific compute cap using the +`CUDA_COMPUTE_CAP=<compute cap>` environment variable. + If any of the above commands errors out, please make sure to update your Cuda version. 2. Create a new app and add [`candle-core`](https://github.com/huggingface/candle/tree/main/candle-core) with Cuda support. |