windows10如何正确的安装tensorflow

开发环境安装(windows 10)

  1. Anaconda3包含python3.7(为了方便虚拟环境及版本管理)
  2. CUDA 10.0
  3. CUDNN
  4. tensorflow 2.0

步骤:

  • 安装Anaconda是傻瓜式安装,详细过程就不讲了
  • cuda的安装

custom(自定义安装)>去掉CUDA选项下的Vistual Studio...>比较Driver components选项下的Display Driver自带的版本号(self)与已安装的版本号(current)=>[if self

  • cudnn(需要对应cuda版本)

首先需要在官网上注册账号后才能下载,将下载的压缩包解压后复制到cuda的安装目录下并将文件夹重命名为cudnn

  • 系统环境配置

将cuda安装好后系统环境变量中已经存在了两条路径,只需添加上另外两条路径到系统环境变量即可(后面两条路径已经省略了cuda的安装目录,真实配置时需要使用绝对路径才行).../cuda/v10.0/extras/CUPTI/libx64.../cuda/v10.0/cudnn/bin,然后保证有4条环境变量并且在最前面,进入cmd输入nvcc -V输出版本号就是安装成功了

  • tensorflow的安装

建议安装在一个虚拟环境中,就算玩坏了直接删除就可以,简单方便,直接给出步骤:

conda create --name py37 python==3.7.0
conda activate py37
pip install tensorflow-gpu==2.0.0
(如果安装cpu版本 pip install tensorflow==2.0.0)
  • 验证

在py37的虚拟环境下导入tensorflow如果不报错就表示成功了

conda activate py37
python
import tensorflow

注意事项

  1. cuda、cudnn、tensorflow的版本一定不能安装错误,去官网对照其中的版本要求
    https://tensorflow.google.cn/install/source#linux下的Linux/macOS的栏目下
  2. 以上只是给出了gpu版本的安装方法,如果是cpu版本的一定要跳过2,3步;其他步骤是一样的。
  3. 本人在安装上述开发环境时解决遇到的一些坑所整理出来的,希望能够解决这个方面有疑惑的同学。

你可能感兴趣的:(windows10如何正确的安装tensorflow)