huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: parse Cuda compute cap from env (#1066)

Browse Source 
* feat: add support for multiple compute caps

* Revert to one compute cap

* fmt

* fix
This commit is contained in:
OlivierDehaene
2023-10-16 16:37:38 +02:00
committed by GitHub
parent 0106b0b04c
commit 75629981bc
4 changed files with 168 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-book/src/guide/installation.md
View File

@ -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.