"docs/source/vscode:/vscode.git/clone" did not exist on "ca302a13ac4aac6a678794c4dc21f8f775728937"
Unverified Commit 04885664 authored by Won-Kyu Park's avatar Won-Kyu Park Committed by GitHub
Browse files

fix cudart*dll for Windows (#1064)

parent 1d709aad
......@@ -30,7 +30,7 @@ from .env_vars import get_potentially_lib_path_containing_env_vars
DYNAMIC_LIBRARY_SUFFIX = { "Darwin": ".dylib", "Windows": ".dll", "Linux": ".so"}.get(platform.system(), ".so")
if platform.system() == "Windows": # Windows
CUDA_RUNTIME_LIBS = ["nvcuda.dll"]
CUDA_RUNTIME_LIBS = ["cudart64_110.dll", "cudart64_12.dll"]
else: # Linux or other
# these are the most common libs names
# libcudart.so is missing by default for a conda install with PyTorch 2.0 and instead
......
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