RTX2080ti显卡+win10+安装Tensorflow-gpu

补充:2018年12月,我又看了一下下面tensorflow的链接,已经有支持CUDA10版本的了,大家直接安装CUDA10就可以了。安装CUDA9.2可能会带来一些问题,就不要安装CUDA9.2了!

原文:

刚买了一台显卡是RTX 2080ti的机器,用来吃鸡。我已经在不同系统上多次搞过tensorflow的安装卸载版本更换等,遇见并解决了了好多问题,所以给这台机器配置TensorFlow我以为是挺简单的事,结果搞了半天才搞定。

 

要是安装tensorflow就先看一下我的另一篇博客:https://blog.csdn.net/macunshi/article/details/80747248

一般显卡情况下基本上都能搞定,但今天要说的是怎么把RTX 2080ti用起来,据我资料搜索结果看本文是全网中文第一篇讲在RTX2080ti上安装tensorflow的。

新机器,我先安装了VS2015,然后安装CUDA9.0结果软件检查环境说 没有合适的硬件,装了CUDA9.0也没用。

RTX 2080ti是英伟达新出的一代显卡,通过进入英伟达控制面板,点击帮助,选择系统信息,再点组件,看到显卡支持CUDA10

然后我就去官网下载CUDA10和匹配的cuDNN,然后安装,顺利进行,再安装python,用pip安装tensorflow-gpu,显示tensorflow安装成功,然后在DOS窗下输入

python
import tensorflow as tf
tf.__version__

结果报错了,没有安装成功!经过查看资料发现tensorflow还不支持CUDA10,那怎么办呢?

经过一番折腾,最后还是搞定了,关键步骤如下,部分细节参考上面那一篇博客:

1.安装VS2015

2.安装python3.6

3.去官网下载CUDA9.2,最新的,还有cuDNN,然后安装上。

    把cuDNN里面的三个文件夹考到

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2下面

    CUDA9.2竟然是支持RTX 2080ti的!

4.可是tensorflow是不支持CUDA9.2的,怎么办?

官方没提供,民间大神来搞定,到这里https://github.com/fo40225/tensorflow-windows-wheel下载符合条件的tensorflow的whl文件,随便找个文件夹下一放。然后按着shift键,右击这个文件夹,选择 在此处打开Powershell,然后在命令窗里输入

pip install *.whl  
//注意*是你刚才下载的whl的文件名称,你自己改过来

完成后再从DOS窗输入最上面的三行代码,正常显示了tensorflow的版本。

现在就可以使用RTX 2080ti跑tensorflow了。

你可能感兴趣的:(深度学习)