Dear PyTorch folks, My apologies, I am a PyTorch newbie and am probably missing something basic. I made a click-to-zoom Mandelbrot set in Google Colab, and am trying to transfer it to Torch, in order to get GPU speed-up. I have selected a GPU runtime and it is showing up in nvidia-smi, and I am pa…

[image] Rajeev_Raizada: complex_tensor.to(device='cuda') z_old.to(device='cuda') z_new.to(device='cuda') abs_val_sq.to(device='cuda') timesteps_in_bound.to(device='cuda') You need to assign the results to a variable: x = x.to(device='cuda') since tensors won’t be moved i…

[image] Moving tensor to cuda Hi, this works, a = torch.LongTensor(1).random_(0, 10).to("cuda"). but this won’t work: a = torch.LongTensor(1).random_(0, 10) a.to(device="cuda") Is this per design, maybe I am simple missing something to convert tensor from CPU to CUDA? …

Thanks! That was what I was missing!

Here’s my working code. [image] Google Colab It computes fairly fast, so I think the thing limiting its speed is the matplotlib rendering in Google Colab. I’m sure that more modern plotting libraries would be faster (plotly, bokey, seaborn etc.), but I haven’t …

Q: Trying to run Mandelbrot on GPU via PyTorch, but not seeing speed-up

lennarteingunia (Lennart Eing) November 27, 2024, 3:47pm 2