cannot support ROCM - Githubissues

ryujaehun / pytorch-gpu-benchmark

Using the famous cnn model in Pytorch, we run benchmarks on various gpu.

MIT License

227 stars 85 forks source link

Closed znsoftm closed 3 years ago

znsoftm commented 3 years ago

With amd GPU, it can only use CPUs to do testing tasks.

Can you guys support AMD GPU with ROCM?

gutama commented 3 years ago

hi, i just benchmark my machine rtx 2060, produces all csv result, but I cannot produce the graph, attached the result, how to interpret the result?

thank you

ryujaehun commented 3 years ago

@znsoftm this rocm docker will help you.

znsoftm commented 3 years ago

@znsoftm this rocm docker will help you.

thanks. I mean DGL runs with rocm. It has native code linked to cuda native API, so we need a transplant from cuda to rocm for DGL.