Tensorflow doesn't seem to see my gpu

前端 未结 6 2038
面向向阳花
面向向阳花 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:37

    If you are using conda, you might have installed the cpu version of the tensorflow. Check package list (conda list) of the environment to see if this is the case . If so, remove the package by using conda remove tensorflow and install keras-gpu instead (conda install -c anaconda keras-gpu. This will install everything you need to run your machine learning codes in GPU. Cheers!

    P.S. You should check first if you have installed the drivers correctly using nvidia-smi. By default, this is not in your PATH so you might as well need to add the folder to your path. The .exe file can be found at C:\Program Files\NVIDIA Corporation\NVSMI

提交回复
热议问题