Cuda Installed but not Available to Python

I used the default installation and get the following from nvcc --version

C:\Users\David>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Mon_Apr__3_17:36:15_Pacific_Daylight_Time_2023
Cuda compilation tools, release 12.1, V12.1.105
Build cuda_12.1.r12.1/compiler.32688072_0

However I can’t train my pytorch models using cuda


Could someone please advise?