-
Notifications
You must be signed in to change notification settings - Fork 64
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
RuntimeError: Error building extension 'emd_ext' #45
Comments
Hi, did you solve it? I encouter the same issure |
export TORCH_CUDA_ARCH_LIST="7.5" before training |
|
if not use, please writer export TORCH_CUDA_ARCH_LIST="7.5" in terminal. The "8.0" usually represents the NVIDIA Ampere architecture, which includes a range of GPU models, such as the NVIDIA A100.You can use different numbers to match the version of your GPU. |
|
How to solve it locally ?
The text was updated successfully, but these errors were encountered: