window10上安装python+CUDA+CuDNN+TensorFlow

纵饮孤独 提交于 2020-04-06 08:53:09

确认显卡支持 cuda

首先确认显卡是否是 英伟达 NVIDIA 的,当然 AMD 也支持,但是不常用;

NVIDIA 显卡有 GTX Geforce、Quadra 和 Tesla 三大系列,然后到如下网站查看是否支持 Cuda

https://developer.nvidia.com/cuda-gpus 

点击如下链接即可查看每个系列支持 cuda 的显卡版本

 

安装显卡驱动

在安装CUDA过程中,你可能各种尝试,把显卡驱动整坏了,可以重新安装;    【我就是这样,然后死活装不上 cuda,后来重装驱动,搞定】

在 https://www.nvidia.cn/Download/index.aspx?lang=cn 获取自己显卡的最新驱动版本;

下载,双击安装;

 

安装 cuda 驱动

访问CUDA的下载网站:https://developer.nvidia.com/cuda-toolkit,可以看到CUDA目前的最新版本,可以通过选择下面的“Legacy Releases”链接来下载旧版本的驱动;

建议选择自定义安装,然后只勾选cuda,建议安装在默认文件路径;

把 安装路径下的 bin 目录设为环境变量:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

 

安装 cudnn 库

CuDNN库(The NVIDIA CUDA® Deep Neural Network library)是CUDA针对深度神经网络的更新包,TensorFlow会使用它用于加速NVidia GPU上的深度学习。可以从这里下载,见:https://developer.nvidia.com/cudnn

首先要注册一个NVidia开发者帐号,它是免费的。登录后,您会看到各种CuDNN下载;

cuda 与 cudnn 版本要严格对应,对应关系见 https://developer.nvidia.com/rdp/cudnn-archive

 

下载下来的是一个包含了几个文件夹的ZIP文件,每个文件夹包含CuDNN文件(一个DLL,一个头文件和一个库文件)。找到你的CUDA安装目录,这里应该是这样的:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

可以看到从ZIP文件的目录也在这个目录,即有一个bin、一个include,一个lib等。将文件从ZIP复制到相关的目录。 

比如把cudnn64_7.dll文件拖拽到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin目录,其它相似。

 

安装 tensorflow-gpu

首先安装 python3.6

具体过程参考我的其他博客,这里只记录重点

之前我讲过 python 安装不推荐 64 位的,所以我安装的是 32 位的,但是死活装不上 tf,报错如下

Could not find a version that satisfies the requirement tensorflow-gpu(from versions: )
No matching distribution found for  tensorflow-gpu

后来换成了 64 位,可以了,我也是醉了

 

然后使用 pip 安装 tensorflow

python3 -m pip install tensorflow-gpu

如果安装很慢,可以使用国内镜像源

python3 -m pip install tensorflow-gpu -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

其中 --trusted-host pypi.douban.com 的作用是获得ssl证书的认证

 

验证是否安装成功

import tensorflow as tf
print('GPU', tf.test.is_gpu_available())        # GPU True

显示 GPU 可用,成功;

 

如果CUDA驱动程序有错误,就可能会显示 cudart64_XX.dll 失败,其中XX是版本号。

ImportError: Could not find 'cudart64_100.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 10.0 from this URL: https://developer.nvidia.com/cuda-90-download-archive

按照提示,要下载 cuda 10.0,tf1.13 以上的版本需要 cuda 10.0 版本

 

如果CUDA驱动程序正确,但CuDNN驱动程序有错误,就可能会显示说 cudnn64_X.dll 缺少什么东西,其中X是一个版本号。 

 

 

 

参考资料:

https://blog.csdn.net/hzk594512323/article/details/86082852  Python之pip安装失败----Could not find a version that satisfies the requirement xxxx(from versions: )

### 下面是 cuda 安装教程 

https://www.cnblogs.com/touch-skyer/p/8367706.html  window10上安装python+CUDA+CuDNN+TensorFlow      【主要参考】

https://blog.csdn.net/u014695788/article/details/93246548  python CUDA配置

https://zhuanlan.zhihu.com/p/29841665  用GPU加速深度学习: Windows安装CUDA+TensorFlow教程

https://www.360kuai.com/pc/9290487ad261e50da?cota=4&tj_url=so_rec&sign=360_57c3bbd1&refer_scene=so_1  tensorflow版本与cuda cuDNN版本对应关系

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!