Can you maybe print action_size. Just so I can see what this list contains.
Just do the same as you did with



 dani_kal:

print(type(action_size))


but remove the type() so it’s just print(action_size) and post the result.
But judging from this error



 dani_kal:

RuntimeError: CUDA error (35): CUDA driver version is insufficient for CUDA runtime version


it actually should have worked, but there might be a problem with CUDA or graphics driver version, or you are trying to use CUDA when you are supposed to use cpu (e.g. if you are on a cpu installation of pytorch or don’t have a compatible gpu)