Skip to content

Commit

Permalink
check gpu for PyTorch
Browse files Browse the repository at this point in the history
  • Loading branch information
EndermanPC committed Jul 9, 2024
1 parent 9a0287b commit 1a535b4
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
1 change: 1 addition & 0 deletions Create.cs
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,7 @@ private async Task InstallPackageDependencies(string language)
break;

case "Python (PyTorch)":
CheckForDiscreteGPU();
packageRequirements.Add("torch");

if (comboBox1.Text == "Text generation (LSTM)")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
[assembly: System.Reflection.AssemblyCompanyAttribute("Lithicsoft Trainer Studio")]
[assembly: System.Reflection.AssemblyConfigurationAttribute("Debug")]
[assembly: System.Reflection.AssemblyFileVersionAttribute("1.0.0.0")]
[assembly: System.Reflection.AssemblyInformationalVersionAttribute("1.0.0+c021e2a6272dc6655cd94992eeb67695442a18ef")]
[assembly: System.Reflection.AssemblyInformationalVersionAttribute("1.0.0+9a0287bd40395f9f38b34f5c5a09c786461a5176")]
[assembly: System.Reflection.AssemblyProductAttribute("Lithicsoft Trainer Studio")]
[assembly: System.Reflection.AssemblyTitleAttribute("Lithicsoft Trainer Studio")]
[assembly: System.Reflection.AssemblyVersionAttribute("1.0.0.0")]
Expand Down
Original file line number Diff line number Diff line change
@@ -1 +1 @@
db5e0d0fd2f509c421af868cd512753bebdcf342687d70a36848a6f29fa9e17e
a33c0c9f7ab1868e17dbf5d1d47c39966de876e18ee1ef895ecd528b2925d409

0 comments on commit 1a535b4

Please sign in to comment.