-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Modified the project to support running of TensorFlow on GPU on Windows. #4270
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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; | ||
|
|
||
|
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please revert this file.
|
|
||
| <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.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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
codemzs
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
![]()
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