Win10系统安装TensorFlow-GPU过程及遇到的坑

win10安装TensorFlow-gpu版本网上的资料很多,这里大致讲下我安装的过程以及在这个过程遇到的一些问题。

本人安装的各个软件的版本号如下:

  • Anaconda3
  • Python3.6.5
  • Cuda10.1
  • Cudnn7.6
  • VS2013(VS版本好像没有硬性要求,只要有VS就行,我是这么认为的,网上说必须安装对应版本的VS,但我只装了2013版本,最后还是安装成功了。主要有个软件必须要装,就是Microsoft Visual C++ Redistributable for Visual Studio 2015, 2017 and 2019,Tensorflow 2.1.0必须有这个,我装了两次没成功就因为没安装这个,安装完重启后居然成功了。下载地址:https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads)

参考信息:https://github.com/tensorflow/tensorflow/releases

Win10系统安装TensorFlow-GPU过程及遇到的坑_第1张图片

  • Tensorflow-gpu 2.1.0
  • 显卡:GTX1650

安装步骤

1.下载和安装Anaconda。这步比较简单直接官网上下载安装就行,官网地址:https://www.anaconda.com/distribution/

这个说下安装Anaconda时,如果没有勾选将Anaconda添加到环境变量中,安装完之后需要手动添加。

2.下载和安装CUDA,CUDA下载地址:https://developer.nvidia.com/cuda-downloads,要查看自己显卡所支持的CUDA版本。打开NVIDIA控制面板,帮助>系统信息>组件里就可以看到显卡所支持的CUDA版本 

我的是支持10.2的,所以下载CUDA版本不要高于这个版本就行,低于应该没有关系。这里有个坑:就是虽然我的电脑支持CUDA10.2版本,但是TensorFlow的版本不支持啊,不知道在我写这篇文章的时候有没有支持,下面给出TensorFlow1.2到2.1本版本各个CUDA和cuDNN所对应的版本集合。所以一定要选择TensorFlow版本对应的CUDA,cuDNN版本,一定要选择正确。这里一定要注意,CUDA版本动辄2G,下错版本会很耗时的。

Win10系统安装TensorFlow-GPU过程及遇到的坑_第2张图片

3.下载和安装cuDNN,下载与CUDA版本对应的版本。这个是深度神经网络的GPU加速库。下载需要注册,可能稍微费时一点。这里也一定要选择与CUDA对应的版本。

下载地址:https://developer.nvidia.com/rdp/cudnn-download 。

可以参考下面这个网址去安装CUDA和cuDNN以及配置环境,相对来说CUDA和cuDNN还是比较好安装的。

http://baijiahao.baidu.com/s?id=1664131820191905649&wfr=spider&for=pc

4.CUDA和cuDNN安装好之后,环境配置完,就可以安装tensorflow-gpu了。首先打开Anaconda下的Anaconda Prompt,先创建一个环境变量,安装不同版本的python过程:

GPU版本:

conda create --name tensorflow-gpu python=3.6.5 

CPU版本:

conda create --name tensorflow python=3.6.5 

如果要卸载环境:

conda remove --name tensorflow-gpu python=3.6.5

然后输入activate tensorflow-gpu,进入到创建好的环境。进入环境之后就可以安装tensorflow-gpu了,可以直接输入:

 pip install tensorflow-gpu

但是这样会很慢,所以建议先下载好tensorflow-gpu。下载地址:https://pypi.org/project/tensorflow-gpu/#files

我在下载完之后安装还遇到一个问题,就是安装的较慢,有时安装到一半会报如下错误:

Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None))

具体的解决方法:使用这种方式安装,pip install pymysql -i url( 其他pip源),pymysql是要安装的或者下载下来的wheel文件,url可以使用国内几个pip源:

阿里云 http://mirrors.aliyun.com/pypi/simple/

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/

豆瓣(douban) http://pypi.douban.com/simple/

清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/
借鉴网址:https://blog.csdn.net/nahanai/article/details/96435128

安装过程没什么问题的话,至此tensorflow-gpu就安装完成了。

验证

下面就是对TensorFlow进行验证,在编译器中输入以下代码:

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

验证结果1:

Win10系统安装TensorFlow-GPU过程及遇到的坑_第3张图片

或者输入以下代码:

import tensorflow as tf

print(tf.__version__)
print(tf.test.is_gpu_available())

验证结果2:

如果出现以上结果就说明tensorflow-gpu安装成功了。

问题

1.安装过程中要查看显卡是否支持CUDA,如果支持要看显卡支持的版本,一定要下载对应的版本,不能下载超过自己显卡支持的版本号。

2.要查看安装的TensorFlow版本号跟CUDA,cuDNN版本号是否对应,可以参考上面的表格。

3.一定要下载Microsoft Visual C++ Redistributable。这个是我一直没安装成功的主要原因,如果没装这个,会报错:Failed to load the native TensorFlow runtime。

总结

总体来说,现在安装TensorFlow-GPU还算比较简单,可能会遇到一些坑,但通过百度/Google都能解决,以上是我在安装过程中遇到的问题,希望对大家有所帮助。可能不同的人遇到不同的坑,仅以此来跟大家分享我所遇到的问题。

你可能感兴趣的:(TensorFlow)