Windows 10安装TensorFlow之TensorFlow篇

Windows 10安装TensorFlow之TensorFlow篇

TensorFlow使用GPU版本效率会更高,本文将讲解TensorFlow-GPU版本的安装,由TensorFlow官网可知安装GPU版本时所需要的硬件和软件的requirements,它需要NVIDIA的GPU(硬件)和NVIDIA 驱动、CUDA、CUDNN(软件),其他可以不用。

接下来将讲解TensorFlow的安装。

版本对应

请看Windows 10安装TensorFlow之CUDA篇

安装

  1. CUDA安装
    请看Windows 10安装TensorFlow之CUDA篇
  2. 安装TensorFlow
    1. 方法一:

      PYPI官网–> 搜索 tensorflow-gpu(一般第一个就是了)–> 点击Release history–> 点击你要的版本–> 点击Download files–> 下载找到你的版本下载的版本

    2. 方法二:直接安装

      pip install tensorflow-gpu==对应版本

      eg: pip install tensorflow-gpu==1.15.0rc3

      注:建议安装Anaconda+Pycharm使用更佳

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章