Jupyter kernal crash in vscode while using tensorflow GPU

Keras
,
1

I am tryping to train a deeplearning model and when i start the training process the kernal crashes

image
image1131×417 19.9 KB

note that tensorflow is able to detect nvidia GPU

config:
hardware-
GTX 1650 4gb ram
16gb ram
ryzen 5 4600h

environmrnt config:
conda environment with python 3.9.19 and tensorflow 2.10.1

image
image856×358 10.4 KB

log:

image
image1887×406 51.7 KB

i tried the solution given here: Kernel crashes when using tensorflow · microsoft/vscode-jupyter Wiki · GitHub

3

Hi @Vivek_Choudhry, Could please let us know which OS you are facing this issue. Also I can see that you are using CUDA 12.4 but as per test build configuration Tensorflow 2.10 supports CUDA 11.2. could please try by downgrading your cuda version. Thank You.