How can I make tensorflow run on a GPU with capability 2.x?

前端 未结 3 636
既然无缘
既然无缘 2020-12-03 02:37

I\'ve successfully installed tensorflow (GPU) on Linux Ubuntu 16.04 and made some small changes in order to make it work with the new Ubuntu LTS release.

However, I

3条回答
  •  伪装坚强ぢ
    2020-12-03 03:18

    Sep.2017 Update: No way to do that without problems and pains. I've tried hard all the ways and even apply below trick to force it run but finally I had to give up. If you are serious with Tensorflow just go ahead and buy 3.0 compute capability GPU.

    This is a trick to force tensorflow run on 2.0 compute capability GPU (not officially):

    1. Find the file in Lib/site-packages/tensorflow/python/_pywrap_tensorflow_internal.pyd (orLib/site-packages/tensorflow/python/_pywrap_tensorflow.pyd)
    2. Open it with Notepad++ or something similar

    3. Search for the first occurence of 3\.5.*5\.2 using regex

    4. You see the 3.0 before 3.5*5.2, change it to 2.0

    I changed as above and can do simple calculation with GPU, but get stuck with strange and unknown issues when try with practical projects(those projects run well with 3.0 compute capability GPU)

提交回复
热议问题