Hey guys
Im facing the following error on moving my model parameters and state to the GPU -

torch.backends.cudnn.CuDNNError: 8: b’CUDNN_STATUS_EXECUTION_FAILED’

(can produce stacktrace if req.) with the following configuration -
cuda90                    1.0                  h4c72538_0    peterjc123
cudatoolkit               9.0                           1
pytorch                   0.3.1           py36_cuda90_cudnn7he774522_2  [cuda90]  peterjc123

However, with -
cudatoolkit               8.0                           4
pytorch                   0.3.1           py36_cuda80_cudnn6he774522_2    peterjc123

the execution runs with just a minor bump in the road. (incompatible compilation warnings)
Following are my hardware specs -

Card: RTX 2080S
CUDA: 10.1, CuDNN 7.6.5

Any help would be appreciated. Thanks!
PS This is my first post on the Torch forum. It would be fine if minor formatting suggestions are commented too.