Skip to content

ROCm support #55

Answered by jongwook
mjsulik asked this question in Q&A
Sep 23, 2022 · 11 comments · 18 replies
Discussion options

You must be logged in to vote

We haven't tested ROCm, but from this documentation it seems that you can keep using cuda if the ROCm version is properly installed.

PyTorch for HIP intentionally reuses the existing torch.cuda interfaces. This helps to accelerate the porting of existing PyTorch code and models because very few code changes are necessary, if any.

Replies: 11 comments 18 replies

Comment options

You must be logged in to vote
5 replies
@mjsulik
Comment options

@DrMemoryFish
Comment options

@marcin-przywoski
Comment options

@mjsulik
Comment options

@EchedelleLR
Comment options

Answer selected by jongwook
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
1 reply
@GrahamboJangles
Comment options

Comment options

You must be logged in to vote
4 replies
@colemannugent
Comment options

@jp-berg
Comment options

@Whitesttax
Comment options

@gnevan
Comment options

Comment options

You must be logged in to vote
7 replies
@Whitesttax
Comment options

@mjsulik
Comment options

@RaymondTheBrave0
Comment options

@RaymondTheBrave0
Comment options

@webdevjosue
Comment options

Comment options

You must be logged in to vote
1 reply
@BernardBurke
Comment options

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet