Running it on a CPU (since I don't have CUDA-enabled GPU)

[tsuijenk]

Hello,

Since I don't have a CUDA-enabled GPU, I have to run it on my CPU for now. I am following this tutorial: https://pytorch.org/tutorials/recipes/recipes/save_load_across_devices.html

I would like some confirmation to see whether or not I am doing it correctly. Any advice and guidance is appreciated! :)

** refers to the newly updated lines for running it on CPU.

In Load Model and Parameters,

ckpt_file = 'sample_data/colab_fit_experimental_rc/ckpt.pt' #@param {type:"string"}

**device = torch.device('cpu')**

ckpt = torch.load(ckpt_file, map_location=**device**)

In Setup Preprocessing Pipeline,

# hardware
#device = 'cuda:0'
**device = 'cpu'**

I realized that I also have to adjust the batch size from "auto" to some more realistic numbers. Any advice? :)

[Haydnspass]

Dear @tsuijenk
which notebook are you referring to? Training or Fit?
In either cases there is this line in the first few cells

Setting device = 'cpu' should be sufficient.