search

aieater / rocm_pytorch_informations

The official page of ROCm/PyTorch will contain information that is always confusing. On this page we will endeavor to describe accurate information based on the knowledge gained by GPUEater infrastructure development.
https://www.gpueater.com/
88 stars 9 forks source link

Some rocm queries #10

Open sbmaruf opened 4 years ago

sbmaruf commented 4 years ago

It seems AMD is catching up with Nvidia. I am thinking about considering AMD GPUs instead of Nvidia because of the scarcity of 30 series. Need Some infos.

  1. How is the fp16 performance for ROCm pytorch?
  2. What are the known issues with the rocm_pytorch?
  3. Is there any performance benchmark?
Dan-RAI commented 3 years ago

If you don't want to waste a lot of time trying to get ROCm running in a configuration you want, stay with NVIDIA ... (even getting standard DL frameworks running is a pain)