RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'mat1'

Question

RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'mat1'

Jeffrey28 opened this issue 5 years ago · 9 comments

I run the Cart_Pole.py and got the error.

Answer 1 · 2019-10-25T07:50:41.000Z

And if I choose “use_GPU" is True. It works well. However, when it ran with A3C, I got the error: Runtime Error: CUDA error (3): initialization error. So, do you have any suggestions for me?

Answer 2 · 2019-10-28T12:31:03.000Z

yeah,i have the same problem, when i run python3 results/Cartpole.py

Answer 3 · 2019-10-28T14:11:28.000Z

GPU doesn't work with A3C because A3C uses multiprocessing - try turning off the GPU to see if it works. For small games anyway it is faster to run it with the GPU turned off.

Answer 4 · 2019-10-29T11:58:34.000Z

@p-christ but when i turn off the gpu. it will display
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'mat1'

Answer 5 · 2019-10-30T08:43:00.000Z

yeah

@p-christ but when i turn off the gpu. it will display
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'mat1'

yeah, the same problem.

Answer 6 · 2019-10-30T12:17:26.000Z

Have you also set: config.use_GPU = False ?

Answer 7 · 2019-10-30T14:28:51.000Z

Have you also set: config.use_GPU = False ?

Yeah, I have set.

Answer 8 · 2019-10-30T14:36:50.000Z

So got the RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'mat1'

Answer 9 · 2019-12-04T06:59:53.000Z

Have you solved this problem yet?I have the same problem when I run results/Cartpole with any agent.