running on GPU

[SilentReader99]

i'm trying to train my model using GPU on HPC, but it always run ob CPU, can you please show me the specific syntax to tell the computer to run it on GPU and and we manage how many GPU or CPU that we can assign to run the code? thank you so much

[wtheune]

I am having the same issue and have not been able to find a solution yet. HPC defaults to GPU with no way to specify to utilize CPU instead. I am getting an error that several packages are using version that are incompatible with our HPC GPU and I'm afraid upgrading the packages will break something.