Not working with cuda device.

Question

Not working with cuda device.

hemangjoshi37a opened this issue 2 years ago · 4 comments

hemangjoshi37a commented 2 years ago

The whole code seems to be running on CPU only and not on CUDA.

How to add this to instructGOOSE?

Answer 1 · 2023-03-28T22:54:30.000Z

@hemangjoshi37a check it out https://www.tutorialspoint.com/how-to-move-a-torch-tensor-from-cpu-to-gpu-and-vice-versa

Answer 2 · 2023-03-29T06:17:56.000Z

I tried calling .to(device) method on instructGOOSE but it gives me error that there are mix type of devices are selected in the process. It means there needs to be library level changes to make it work with GPUs.

Answer 3 · 2023-03-30T04:18:09.000Z

@hemangjoshi37a Thanks for let me know. I'm working on support it soon

Answer 4 · 2023-03-30T04:21:19.000Z

@xrsrke actually I wanted to create a generalized system that implements instruct over any machine learning model. Like textClassificationInstruct, textNERinstruct, imageObjectInstruct, imageDepthInstruct, imageClassificationInstruct.