Tensorflow doesn't seem to see my gpu

前端 未结 6 2037
面向向阳花
面向向阳花 2020-12-09 01:19

I\'ve tried tensorflow on both cuda 7.5 and 8.0, w/o cudnn (my GPU is old, cudnn doesn\'t support it).

When I execute device_lib.list_local_devices(),

6条回答
  •  死守一世寂寞
    2020-12-09 01:44

    I came across this same issue in jupyter notebooks. This could be an easy fix.

    $ pip uninstall tensorflow
    $ pip install tensorflow-gpu
    

    You can check if it worked with:

    tf.test.gpu_device_name()
    

    Update 2020

    It seems like tensorflow 2.0+ comes with gpu capabilities therefore pip install tensorflow should be enough

提交回复
热议问题