Update win_install.bat

This commit is contained in:
Saifeddine ALOUI 2023-10-16 15:03:20 +02:00 committed by GitHub
parent d2a8774e31
commit 0afe7a1e91
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -45,7 +45,7 @@ set gpuchoice=%gpuchoice:~0,1%
if /I "%gpuchoice%" == "A" ( if /I "%gpuchoice%" == "A" (
set "PACKAGES_TO_INSTALL=python=3.10 cuda-toolkit ninja git" set "PACKAGES_TO_INSTALL=python=3.10 cuda-toolkit ninja git"
set "CHANNEL=-c nvidia/label/cuda-11.7.0 -c nvidia -c conda-forge" set "CHANNEL=-c nvidia/label/cuda-11.8.0 -c nvidia -c conda-forge"
) else if /I "%gpuchoice%" == "B" ( ) else if /I "%gpuchoice%" == "B" (
set "PACKAGES_TO_INSTALL=python=3.10 rocm-comgr rocm-smi ninja git" set "PACKAGES_TO_INSTALL=python=3.10 rocm-comgr rocm-smi ninja git"
set "CHANNEL=-c conda-forge" set "CHANNEL=-c conda-forge"