Solved by mbullo in post #2 
              
              
                Hi Timangar. I’m not familiar with c++ but looking at the docs torch::load() should call load_from which allows to specify the device through the parameter device. Maybe this can help!