Skip to content

Update torch to version 2.0.2#41

Open
jbargu wants to merge 2 commits into
clarinsi:masterfrom
jbargu:jg_update_torch_version_to_2
Open

Update torch to version 2.0.2#41
jbargu wants to merge 2 commits into
clarinsi:masterfrom
jbargu:jg_update_torch_version_to_2

Conversation

@jbargu
Copy link
Copy Markdown

@jbargu jbargu commented Aug 7, 2023

Desciption

The current torch version supports CUDA version 10.2, which supports CUDA capabilities up to sm_70. Newer GPUs are not supported. The PR upgrades torch version to 2.0.2, allowing the use of newer GPUs.

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.

1 participant