• Yi Zhang's avatar
    enable windows cuda unit tests (#4421) · 932ca5a3
    Yi Zhang authored
    
    
    * enable cuda for unittest
    
    * add one check
    
    * rename
    
    * minor fix
    
    * add more check
    
    * add cuda envs
    
    * add more cuda envs
    
    * add driver update
    
    * update vs integration
    
    * get cuda version
    
    * merge
    
    * add envs in install.sh
    
    * reduce some change and add comments
    
    * minor improve
    
    * fix typo
    
    * exit if driver update failed
    
    * fix lint
    
    * fix lint
    
    * Avoid catching exception to show error message
    
    * try 11.1
    
    * try 11.1
    
    * "try 11.1"
    
    * set the downloaded gpu driver dlls only for 10.2
    
    * check torch.cuda.is_available()
    
    * add display.driver
    
    * Revert "Avoid catching exception to show error message"
    
    This reverts commit 1ce58c50de9fefcc3438d29ff398b941f9b6a757.
    
    * Update .circleci/unittest/windows/scripts/set_cuda_envs.sh
    
    * Update .circleci/unittest/windows/scripts/set_cuda_envs.sh
    
    * Update .circleci/unittest/windows/scripts/set_cuda_envs.sh
    Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    Co-authored-by: default avatarNikita Shulga <nikita.shulga@gmail.com>
    Co-authored-by: default avatarNikita Shulga <nshulga@fb.com>
    932ca5a3
cuda_install.bat 12.2 KB