leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-08-16 21:50:33 +03:00
Code Issues Projects Releases Wiki Activity

refactor(scripts): differentiate PyTorch info log on macOS

Browse Source
This commit is contained in:
louistiti 2024-06-24 08:28:32 +08:00
parent 6c3abf30b0
commit 8dc725a697
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/setup/setup-python-dev-env.js
View File

@ -147,7 +147,12 @@ SPACY_MODELS.set('fr', {
* @see https://docs.nvidia.com/deeplearning/cudnn/latest/reference/support-matrix.html
*/
const installPytorch = async () => {
LogHelper.info('Installing PyTorch with CUDA support...')
const logInfo =
osType === OSTypes.MacOS
? 'Installing PyTorch...'
: 'Installing PyTorch with CUDA support...'
LogHelper.info(logInfo)
try {
// There is no CUDA support on macOS
const commandToExecute =