failed call to cuInit: CUDA_ERROR_UNKNOWN: unknown error

[zcx-language]

I have intalled cuda10.2 and cudnn before for my pytorch env and tensorflow2 env, and they work fine

I have installed jax by running below
pip install --upgrade jax jaxlib==0.1.65+cuda102 -f https://storage.googleapis.com/jax-releases/jax_releases.html
and i have installed cudatoolkit=10.2 and cudnn=7.6 in jax env by running
conda intall cudatoolkit=10.2 cudnn

can you give me some susgestion about the error " failed call to cuInit: CUDA_ERROR_UNKNOWN: unknown error" shown in the image ?

[zcx-language]

ok ,i fix it by run
sudo apt install nvidia-modprobe
and restart

thanks all !