Does TensorFlow or CUDA need to be reinstalled for new GPU?

TensorFlow was working fine when I only had a 1660ti, but I recently installed a 3070 and the 3070 works fine for other stuff, but not TensorFlow.

It will take a good minute to just do – tf.test.gpu_device_name()

import os os.environ["CUDA_VISIBLE_DEVICES"]="0" import tensorflow as tf tf.test.gpu_device_name() 

If CUDA_VISIBLE_DEVICES is changed to “1” (my old card) it will print the device name instantly.

Everything else TensorFlow related will also take super long on the new card.

Is there some box I have to check in the CUDA installation files for the new card to work, or do I need to specially install something to find the new card?

I’m using Jupyter Notebook.

Any help would be appreciated.

submitted by /u/KingGeorge12321
[visit reddit] [comments]

Leave a Reply

Your email address will not be published.