diff --git a/installer_scripts/js/initializeApp.js b/installer_scripts/js/initializeApp.js index 4361fd2..429a6ed 100644 --- a/installer_scripts/js/initializeApp.js +++ b/installer_scripts/js/initializeApp.js @@ -39,7 +39,7 @@ const commonPackages = [ ffmpegPackage, ]; -const cudaPackages = `pytorch[version=${torchVersion},build=py3.10_cuda${cudaVersion}*] pytorch-cuda=${cudaVersion} torchvision torchaudio cuda-toolkit ninja`; +const cudaPackages = `pytorch[version=${torchVersion},build=py3.10_cuda${cudaVersion}.*] pytorch-cuda=${cudaVersion} torchvision torchaudio cuda-toolkit ninja`; const cudaPytorchInstall$ = [ "conda install -y -k", ...commonPackages,