pythonのエラーが直せません

解決したいこと

ここに解決したい内容を記載してください。
pythonのエラーが直せません

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

というエラーが出ました

自分で試したこと

print(torch.cuda.is_available())
を動かしたらtureと出ました

また、環境は
Ubuntu 22.04.1 LTS \n \l
GPU　NVIDIA Corporation GA106 [GeForce RTX 3060 Lite Hash Rate] (rev a1)
NVRM version: NVIDIA UNIX x86_64 Kernel Module 520.61.05 Thu Sep 29 05:30:25 UTC 2022
GCC version: gcc version 11.3.0 (Ubuntu 11.3.0-1ubuntu1~22.04)
Cuda compilation tools, release 11.8, V11.8.89
Python 3.10.6
pip 22.0.2
です。
エラーの直し方をわかる方がいましたら教えてください