File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
StabilityMatrix.Core/Models/Packages Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -181,17 +181,17 @@ public override async Task InstallPackage(
181181 }
182182
183183 var torchIndex = options . PythonOptions . TorchIndex ?? GetRecommendedTorchVersion ( ) ;
184- var isBlackwell =
184+ var isLegacyNvidia =
185185 torchIndex is TorchIndex . Cuda
186- && ( SettingsManager . Settings . PreferredGpu ? . IsBlackwellGpu ( ) ?? HardwareHelper . HasBlackwellGpu ( ) ) ;
186+ && ( SettingsManager . Settings . PreferredGpu ? . IsLegacyNvidia ( ) ?? HardwareHelper . HasLegacyNvidia ( ) ) ;
187187
188188 var config = new PipInstallConfig
189189 {
190190 PrePipInstallArgs = [ "joblib" ] ,
191191 RequirementsFilePaths = requirementsPaths ,
192- TorchVersion = "" ,
193- TorchvisionVersion = "" ,
194- CudaIndex = isBlackwell ? "cu128 " : "cu126 " ,
192+ TorchVersion = " " ,
193+ TorchvisionVersion = " " ,
194+ CudaIndex = isLegacyNvidia ? "cu126 " : "cu128 " ,
195195 RocmIndex = "rocm7.2" ,
196196 ExtraPipArgs =
197197 [
You can’t perform that action at this time.
0 commit comments