I’m trying to get CUDA acceleration working on Tensorflow within WSL I have install the preview drivers for CUDA but when adding two tensors in Tensorflow gives me the following error.
Cuda_prob.txt (7.2 KB)
Can someone verify if my installation of CUDA is correct?
I’m trying to install this outside of Docker in WSL.

1 Like

hi there,

I reviewed your log - what is the error again?

We’re printing a number of diagnostic messages indicating various CUDA libraries being loaded when you started the tensor operation. You asked Python to execute the tensor operation to add the matrices: int(tf.add([1.0, 2.0], [3.0, 4.0])) and TF returned the correct matrix add operation result: tf.Tensor([4. 6.], shape=(2,), dtype=float32)

I was confused and thought cuda wasn’t working.
Thank you @P_Ramarao .