Windows 10安装TensorFlow之TensorFlow篇
TensorFlow使用GPU版本效率会更高,本文将讲解TensorFlow-GPU版本的安装,由TensorFlow官网可知安装GPU版本时所需要的硬件和软件的requirements,它需要NVIDIA的GPU(硬件)和NVIDIA 驱动、CUDA、CUDNN(软件),其他可以不用。
接下来将讲解TensorFlow的安装。
版本对应
请看Windows 10安装TensorFlow之CUDA篇
安装
- CUDA安装
请看Windows 10安装TensorFlow之CUDA篇 - 安装TensorFlow
-
方法一:
PYPI官网–> 搜索 tensorflow-gpu(一般第一个就是了)–> 点击Release history–> 点击你要的版本–> 点击Download files–> 下载找到你的版本下载的版本
-
方法二:直接安装
pip install tensorflow-gpu==对应版本
eg: pip install tensorflow-gpu==1.15.0rc3
注:建议安装Anaconda+Pycharm使用更佳
-