diff --git a/torch_utils/custom_ops.py b/torch_utils/custom_ops.py index 439e445..9d23274 100644 --- a/torch_utils/custom_ops.py +++ b/torch_utils/custom_ops.py @@ -32,6 +32,7 @@ def _find_compiler_bindir(): 'C:/Program Files*/Microsoft Visual Studio/*/BuildTools/VC/Tools/MSVC/*/bin/Hostx64/x64', 'C:/Program Files*/Microsoft Visual Studio/*/Community/VC/Tools/MSVC/*/bin/Hostx64/x64', 'C:/Program Files*/Microsoft Visual Studio */vc/bin', + 'C:/Program Files*/Microsoft Visual Studio/*/Preview/VC/Tools/MSVC/*/bin/Hostx64/x64', ] for pattern in patterns: matches = sorted(glob.glob(pattern))