Commit 4b11eee8 authored by Yi Zhang's avatar Yi Zhang Committed by Facebook GitHub Bot
Browse files

improve cuda installation on windows (#2032)

Summary:
1. stop&disable the windows upgrade that's the major reason of the failure of cuda installation
    https://app.circleci.com/pipelines/github/pytorch/audio/8458/workflows/feb65e3b-1093-4724-b849-1a2ac166f354/jobs/441331
     For more details please check out https://github.com/pytorch/pytorch/issues/64536
2. print the log when the cuda installation fails

Pull Request resolved: https://github.com/pytorch/audio/pull/2032

Reviewed By: mthrok

Differential Revision: D32816145

Pulled By: malfet

fbshipit-source-id: 44a2ef0dd4c43469472a6e518ed64841e2dcd5bb
parent 9114e636
......@@ -193,7 +193,12 @@ if not exist "%SRC_DIR%\temp_build\NvToolsExt.7z" (
echo Installing CUDA toolkit...
7z x %CUDA_SETUP_FILE% -o"%SRC_DIR%\temp_build\cuda"
pushd "%SRC_DIR%\temp_build\cuda"
start /wait setup.exe -s %ARGS%
sc config wuauserv start= disabled
sc stop wuauserv
sc query wuauserv
start /wait setup.exe -s %ARGS% -loglevel:6 -log:"%cd%/cuda_install_logs"
echo %errorlevel%
popd
echo Installing VS integration...
......@@ -222,6 +227,10 @@ set "NVTOOLSEXT_PATH=%ProgramFiles%\NVIDIA Corporation\NvToolsExt\bin\x64"
if not exist "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\bin\nvcc.exe" (
echo CUDA %CUDA_VERSION_STR% installed failed.
echo --------- RunDll32.exe.log
type "%SRC_DIR%\temp_build\cuda\cuda_install_logs\LOG.RunDll32.exe.log"
echo --------- setup.exe.log -------
type "%SRC_DIR%\temp_build\cuda\cuda_install_logs\LOG.setup.exe.log"
exit /b 1
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment