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

improve

Browse Source
This commit is contained in:
Patrick von Platen
2023-08-23 08:40:23 +00:00
parent d4968295a0
commit 34cb9f924f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -99,8 +99,9 @@ Having installed `candle` with Cuda support, you can create the tensors on GPU i
```diff
- let a = Tensor::randn(0f32, 1., (2, 3), &Device::Cpu)?;
- let b = Tensor::randn(0f32, 1., (3, 4), &Device::Cpu)?;
+ let a = Tensor::randn(0f32, 1., (2, 3), &Device::new_cuda(0)?)?;
+ let b = Tensor::randn(0f32, 1., (3, 4), &Device::new_cuda(0)?)?;
+ let device = Device::new_cuda(0)?;
+ let a = Tensor::randn(0f32, 1., (2, 3), &device)?;
+ let b = Tensor::randn(0f32, 1., (3, 4), &device)?;
```
For more advanced examples, please have a look at the following sections.