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

[Jeffrey28]

trafficstars

I run the Cart_Pole.py and got the error.

[Jeffrey28]

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?

[lml519]

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

[p-christ]

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.

[lml519]

@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'

[Jeffrey28]

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.

[p-christ]

Have you also set: config.use_GPU = False ?

[Jeffrey28]

Have you also set: config.use_GPU = False ?

Yeah, I have set.

[Jeffrey28]

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

[ztz695183179]

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