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使用更佳

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