Modified the project to support running of TensorFlow on GPU on Windows.#4270
Modified the project to support running of TensorFlow on GPU on Windows.#4270codemzs merged 4 commits intodotnet:masterfrom bpstark:addGPU
Conversation
Codecov Report
@@ Coverage Diff @@
## master #4270 +/- ##
==========================================
+ Coverage 74.46% 74.47% +<.01%
==========================================
Files 877 877
Lines 153660 153660
Branches 16828 16828
==========================================
+ Hits 114428 114442 +14
+ Misses 34496 34484 -12
+ Partials 4736 4734 -2
|
Codecov Report
@@ Coverage Diff @@
## master #4270 +/- ##
=========================================
Coverage ? 74.47%
=========================================
Files ? 877
Lines ? 153660
Branches ? 16828
=========================================
Hits ? 114442
Misses ? 34484
Partials ? 4734
|
|
|
||
| <ItemGroup> | ||
| <PackageReference Include="Microsoft.ML.Onnx.TestModels" Version="$(MicrosoftMLOnnxTestModelsVersion)" /> | ||
| <PackageReference Include="SciSharp.TensorFlow.Redist" Version="1.14.0" /> |
There was a problem hiding this comment.
1.14.0" [](start = 68, length = 7)
1.14.0" [](start = 68, length = 7)
Can this be made a variable in \machinelearning\build\Dependencies.props #Resolved
| { | ||
| int samples = 0; | ||
|
|
||
|
|
|
|
||
| <ItemGroup> | ||
| <PackageReference Include="Microsoft.ML.Onnx.TestModels" Version="$(MicrosoftMLOnnxTestModelsVersion)" /> | ||
| <PackageReference Include="SciSharp.TensorFlow.Redist-Windows-GPU" Version="1.14.0" /> |
There was a problem hiding this comment.
1.14.0 [](start = 80, length = 6)
1.14.0 [](start = 80, length = 6)
Can this be made a variable in \machinelearning\build\Dependencies.props #Resolved
|
|
||
| <ItemGroup> | ||
| <PackageReference Include="Microsoft.ML.TestModels" Version="$(MicrosoftMLTestModelsPackageVersion)" /> | ||
| <PackageReference Include="SciSharp.TensorFlow.Redist" Version="1.14.0" /> |
There was a problem hiding this comment.
Version="1.14.0" /> [](start = 59, length = 19)
Version="1.14.0" /> [](start = 59, length = 19)
Can this be made a variable in \machinelearning\build\Dependencies.props #Resolved
| <PackageReference Include="Microsoft.ML.Onnx.TestModels" Version="$(MicrosoftMLOnnxTestModelsVersion)" /> | ||
| <PackageReference Include="Microsoft.ML.TestDatabases" Version="$(MicrosoftMLTestDatabasesPackageVersion)" /> | ||
| <PackageReference Include="Microsoft.ML.TestModels" Version="$(MicrosoftMLTestModelsPackageVersion)" /> | ||
| <PackageReference Include="SciSharp.TensorFlow.Redist" Version="1.14.0" /> |
There was a problem hiding this comment.
t" Version="1.14.0" /> [](start = 56, length = 22)
t" Version="1.14.0" /> [](start = 56, length = 22)
Can this be made a variable in \machinelearning\build\Dependencies.props #Resolved
Removed all dependencies of TensorFlow redist from the source projects, and instead added the dependency to the Sample project. Created separate sample project for GPU examples since gpu tensorflow requires cuda, which may not be available on all machines, so it needs to be a separate project. Added documentation for setup as there is now some requirements.
Removed all dependencies of TensorFlow redist from the source projects,
and instead added the dependency to the Sample project.
Created separate sample project for GPU examples since gpu tensorflow requires cuda,
which may not be available on all machines, so it needs to be a separate
project.
Added documentation for setup as there is now some setup requirements to use this API.
In testing on the large flowers data set I was able to see a large improvement in speed, from taking ~720 seconds to train to taking ~156 seconds.
Fixes #4269
Addresses part of the issue in #86