Tensorflow2.1 Windows Anaconda conda 安装教程 GPU CPU

机器学习专家建议:别再用pip下载TensorFlow了,换用Conda

https://zhuanlan.zhihu.com/p/46599887

  • 更快的CPU性能

Conda TensorFlow 包利用了用于深度神经网络或 1.9.0 版本以上的 MKL-DNN 网络的英特尔 Math Kernel Library(MKL),这个库能让性能大幅提升。

相比 pip 安装,使用 Conda 安装后的性能足足提升了 8 倍。这对于仍然经常使用 CPU 训练的人来说,无疑帮助很大。我(Michael Nguyen——译者注)自己平时在把代码放到 GPU 驱动的机器之前,会先使用 CPU 机器跑一遍,使用 Conda 安装 TensorFlow 能大幅加快迭代速度。

MKL 库不仅能加快 TensorFlow 包的运行速度,也能提升其它一些广泛使用的程序库的速度,比如 Numpy、NumpyExr、Scikit-Learn。

  • 简化 GPU 版的安装

Conda 安装会自动安装 CUDA 和 GPU 支持所需的 CuDNN 库,但 pip 安装需要你手动完成。大家都比较喜欢一步到位的吧,特别是下载很多个库的时候。

 

windows下

Anaconda安装

Anaconda 是一个用于科学计算的 Python 发行版,支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包。

1 Anaconda 官网下载

网址:https://www.anaconda.com/download/  ,官网上可以选择各个操作系统的安装包。

我们选择最新版本的安装包下载下来,然后双击下载下来的.exe文件就可以安装了。

Tensorflow2.1 Windows Anaconda conda 安装教程 GPU CPU_第1张图片

Tensorflow2.1 Windows Anaconda conda 安装教程 GPU CPU_第2张图片

 2 Anaconda 镜像下载

Anaconda 安装包可以到 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 下载。

1添加源

选择带有Tensorflow2.1的conda 源,添加源。

  1. conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
  2. conda config --set show_channel_urls yes

Tensorflow2.1 Windows Anaconda conda 安装教程 GPU CPU_第3张图片

2 安装tensorflow

      只需一个命令安装tensorflow2.1,会自动安装cudatoolkit=10.1和cudnn=7.6

     1创建一个新环境 安装tensorflow (CPU)
     输入

  1. conda create --name tf2 python=3.7
  2. activate tf2
  3. conda install tensorflow=2.1

     2创建一个新环境 安装tensorflow (GPU)
     输入

  1. conda create --name tfg python=3.7
  2. activate tfg
  3. conda install tensorflow-gpu=2.1

 

3测试tensorflow是否成功

  1. python
  2. import tensorflow as tf
  3. tf.__version__

Tensorflow2.1 Windows Anaconda conda 安装教程 GPU CPU_第4张图片

你可能感兴趣的:(Tensorflow2.1 Windows Anaconda conda 安装教程 GPU CPU)