Skip to content
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

Run inference with pytorch 1.11 and numpy > 1.20 #344

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

AngusG
Copy link

@AngusG AngusG commented Jun 14, 2024

Running inference with RTX 4090, NVIDIA Driver Version: 545.23.08, CUDA Version: 12.3 throws RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED.

Upgrading pytorch to 1.11.0 resolves the issue.

This PR:

  • quiets a pytorch __floordiv__ deprecation warning.
  • numpy version is unspecified in the requirements.txt, however numpy deprecated aliases to int/float after version 1.20 resulting in an error. The proposed fix is adopted.

AttributeError: module 'numpy' has no attribute 'int'.
np.int was a deprecated alias for the builtin int. To avoid this error in existing code, use int by itself. Doing this will not modify any behavior and is safe. When replacing np.int, you may wish to use e.g. np.int64 or np.int32 to specify the precision. If you wish to review your current use, check the release note link for additional information.
The aliases was originally deprecated in NumPy 1.20; for more details and guidance see the original release note at: https://numpy.org/devdocs/release/1.20.0-notes.html#deprecations

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant