Skip to content

Commit

Permalink
Add NuGet version hint (dotnet#886)
Browse files Browse the repository at this point in the history
Signed-off-by: R0Wi <[email protected]>

Co-authored-by: Michael Sharp <[email protected]>
  • Loading branch information
R0Wi and michaelgsharp authored Mar 19, 2021
1 parent d36c073 commit 8167306
Showing 1 changed file with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -50,11 +50,13 @@ To solve this problem, first we will build an ML model. Then we will train the m
By default this solution uses **CPU** for training and scoring.
But if your machine has a compatible **GPU** available (basically most NVIDIA GPU graphics cards), you can configure the project to use GPU.

> :warning: Make sure you use the correct versions of the NuGet packages listed below. Other versions might be incompatiple with Nvidia CUDA v10.0
#### Using CPU for training or inference/scoring

When using **CPU**, your project has to reference the following redist library:

- `SciSharp.TensorFlow.Redist` (CPU training)
- `SciSharp.TensorFlow.Redist (1.15.0)` (CPU training)

Sample references screenshot in training project using **CPU**:

Expand All @@ -64,9 +66,9 @@ Sample references screenshot in training project using **CPU**:

When using **GPU**, your project has to reference the following redist library (*and remove the CPU version reference*):

- `SciSharp.TensorFlow.Redist-Windows-GPU` (GPU training on Windows)
- `SciSharp.TensorFlow.Redist-Windows-GPU (1.14.0)` (GPU training on Windows)

- `SciSharp.TensorFlow.Redist-Linux-GPU` (GPU training on Linux)
- `SciSharp.TensorFlow.Redist-Linux-GPU (1.14.0)` (GPU training on Linux)

Sample references screenshot in training project using **GPU**:

Expand Down

0 comments on commit 8167306

Please sign in to comment.