CUDA Error: No Device Detected

Hello,

I am trying to run a basic CUDA Python program on my laptop with an NVIDIA GeForce RTX 4060 on WSL2 with drivers I installed a day ago. I have followed all of the setup instructions and have even run a test program to say hello from the GPU which was working. However I get the following error when I try to run a basic script which adds 2 vectors of 10 numbers each:

Traceback (most recent call last):
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/driver.py”, line 254, in ensure_initialized
self.cuInit(0)
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/driver.py”, line 327, in safe_cuda_api_call
self._check_ctypes_error(fname, retcode)
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/driver.py”, line 395, in _check_ctypes_error
raise CudaAPIError(retcode, msg)
numba.cuda.cudadrv.driver.CudaAPIError: [100] Call to cuInit results in CUDA_ERROR_NO_DEVICE

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File “/mnt/c/Users/Adittya Patil/Downloads/test/test.py”, line 20, in
d_a = cuda.to_device(a)
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/devices.py”, line 231, in _require_cuda_context
with _runtime.ensure_context():
File “/usr/lib/python3.10/contextlib.py”, line 135, in enter
return next(self.gen)
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/devices.py”, line 121, in ensure_context
with driver.get_active_context():
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/driver.py”, line 495, in enter
driver.cuCtxGetCurrent(byref(hctx))
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/driver.py”, line 292, in getattr
self.ensure_initialized()
File “/home/adittya-patil/.local/lib/python3.10/site-packages/numba/cuda/cudadrv/driver.py”, line 258, in ensure_initialized
raise CudaSupportError(f"Error at driver init: {description}")
numba.cuda.cudadrv.error.CudaSupportError: Error at driver init: Call to cuInit results in CUDA_ERROR_NO_DEVICE (100)

Would somebody please be able to help me with any additional necessary setup/modifications?