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

Upgrade to libtorch v1.13.1 #106

Merged
merged 3 commits into from Jan 4, 2023

Conversation

mfl28
Copy link
Collaborator

@mfl28 mfl28 commented Jan 3, 2023

Description of the change

  • Upgrades libtorch to 1.13.1 in CMake and Docker setup.
  • Updates available CUDA version options (11.6 and 11.7) for libtorch download via CMake.
  • The CMake setup now allows to use a range of libtorch versions (>= 1.12.0, <= 1.13.1) when CMAKE_PREFIX_PATH is used.
  • Sets linear regression tutorial data dtypes explicitly to float (int-type caused a type-mismatch error with latest libtorch). Also the tutorial now uses the GPU if available - same as other tutorials.
  • Updates colab notebook to work with latest libtorch versions. A more recent CUDA version (v11.6 - Colab has incompatible v11.2 by default which causes compilation errors) and more recent g++ are now installed before building the tutorials.

Type Of Change

  • New Feature

Related Issues

Closes #105 .

Development & Code Review

  • cpplint rules passes locally (run cmake -P cpplint.cmake)
  • CI is passing
  • Changes have been reviewed by at least one of the maintainers

@mfl28 mfl28 self-assigned this Jan 3, 2023
@prabhuomkar prabhuomkar merged commit 7a5d057 into prabhuomkar:master Jan 4, 2023
@mfl28 mfl28 deleted the upgrade-libtorch-v1.13.1 branch January 4, 2023 07:53
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.

[feature] Upgrade to libtorch v1.13.1
2 participants