This commit is contained in:
cocktailpeanut
2024-08-09 09:47:08 -04:00
parent 1bf6769c90
commit efe125314d

View File

@@ -37,8 +37,8 @@ module.exports = {
"params": {
"venv": "{{args && args.venv ? args.venv : null}}",
"path": "{{args && args.path ? args.path : '.'}}",
//"message": "pip install torch==2.4.0 torchvision torchaudio==2.4.0"
"message": "pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu"
"message": "pip install torch==2.3.1 torchvision torchaudio==2.3.1"
//"message": "pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu"
}
},
// linux nvidia