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

Apply suggestions from code review

Browse Source
This commit is contained in:
Patrick von Platen
2023-08-23 13:33:45 +02:00
committed by GitHub
parent c98d3cfd8b
commit 1f58bdbb1d
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -31,7 +31,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
} }
``` ```
`cargo run` should display a tensor of shape `Tensor[[2, 4], f32]` `cargo run` should display a tensor of shape `Tensor[[2, 4], f32]`.
Having installed `candle` with Cuda support, simply define the `device` to be on GPU: Having installed `candle` with Cuda support, simply define the `device` to be on GPU:

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

@ -3,7 +3,7 @@
**With Cuda support**: **With Cuda support**:
1. First, make sure that Cuda is correctly installed. 1. First, make sure that Cuda is correctly installed.
- `nvcc --version` should print your information about your Cuda compiler driver. - `nvcc --version` should print information about your Cuda compiler driver.
- `nvidia-smi --query-gpu=compute_cap --format=csv` should print your GPUs compute capability, e.g. something - `nvidia-smi --query-gpu=compute_cap --format=csv` should print your GPUs compute capability, e.g. something
like: like:
@ -14,7 +14,7 @@ compute_cap
If any of the above commands errors out, please make sure to update your Cuda version. 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 2. Create a new app and add [`candle-core`](https://github.com/huggingface/candle/tree/main/candle-core) with Cuda support.
Start by creating a new cargo: Start by creating a new cargo: