win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程
安装环境:win10+Anaconda4.8.3 + Python3.7.7
安装版本:tensorflow1.14.0+CUDA10.0+cuDNN7.6.5
Anaconda安装
Anaconda是一个开源的Python发行版本,其包含了Python、conda等180多个常用科学计算包及其依赖项。相比单独安装Python主程序,安装Anaconda后不需要再单独安装大部分的第三方库。
Anaconda官网下载地址:https://www.anaconda.com/
Tensorflow-GPU版本安装
运行Anconda Prompt进入命令行
- 检查Anaconda的版本:
conda --version
- 检查Anaconda中安装的环境变量:
conda info --envs
- 检查当前可使用的Python版本:
conda search --full-name python
- 创建Anaconda内置的Python版本解析器:
安装GPU版本:conda install --name tensorflow-gpu python=3.7
安装CPU版本:conda install --name tensorflow python=3.7
- **tensorflow环境:
activate tensoflow
- 检查Python的版本:
- 退出当前环境:
conda deactivate
- 检查当前Anaconda创建的环境:
conda info --envs
(如果需要删除Anconda中创建的环境:conda remove --name tensorflow-gpu --all
) - 查询可供安装的tensorflow版本:
conda search --full-name tensorflow
- 安装tensoflow-GPU版本:
pip install tensorflow-gpu==1.14.0
(也可以安装完CUDA和cuDNN后再安装tensorflow)
安装tensorflow-CPU版本:pip install tensorflow==1.14.0
CUDA和cuDNN版本选择
-
查看本设备的CUDA驱动适配版本
进入英伟达控制面板,点击帮助->系统信息,可以查看GPU的完整版本信息
点击组件,可以查看本设备GPU所支持的CUDA版本(可看到本机支持CUDA10.2及以下版本) -
查看tensorflow适配的CUDA版本
在tensorflow官网https://www.tensorflow.org/install/source_windows#gpu可以查询tensoflow各版本对应的CUDA和cuDNN版本,切记版本对应安装!!!(比如安装的tensorflow-gpu版本是v1.14.0,那么CUDA应该装v10.0,cuDNN应该装v7.6.5)
CUDA下载安装
CUDA是NVIDIA推出的只能用于自家GPU的并行计算框架,只有安装该框架才能够进行复杂的并行计算。主流的深度学习框架也都是基于CUDA进行GPU并行加速的。
CUDA官网下载地址: https://developer.nvidia.com/cuda-downloads
- 推荐选择自定义安装
- 推荐默认安装路径
- 检查CUDA是否安装成功以及版本信息:
cmd输入nvcc --version或nvcc -V
cuDNN下载安装
cuDNN是针对深度卷积神经网络的GPU加速库。
cuDNN官网下载地址:https://developer.nvidia.com/cudnn
注意:必须选择和你安装的CUDA相匹配的版本!!!
- 把下载的cudnn-10.0-windows10-x64-v7.6.5.32_2.zip压缩包解压
- 把上面3个文件夹复制到CUDA安装目录下面(直接拖拽粘贴即可,没有重复的文件)
- 安装完成后,进入系统环境变量中添加环境变量
此电脑->属性->高级系统设置->环境变量->系统变量->选择Path进入编辑
添加如下图两个新的环境变量(所以一定要记住CUDA的安装路径!!!v10.0为CUDA版本号,根据下载的版本进行更改) - 检查安装结果
Hello,Tensorflow!
在Anaconda的Jupyter Notebook中运行Helloworld程序: