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

前端 未结 3 645
既然无缘
既然无缘 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:21

    I found it how to install Tensorflow-gpu on a compute capability 2.1 NVIDIA GeForce 525M for python ,the trick is simple use a archived version of tensorflow, I used 1.9.0 The python command for package using PIP is pip install tensorflow-gpu==1.9.0 and cuDNN version is 7.4.1

提交回复
热议问题