一.配置要求
window版本的tf是用不成python 2.X,至少是python 3.5,看官方要求吧,https://tensorflow.google.cn/install/source_windows,如下所示:
所以总结下就是:
- Python 3.5 or 3.6
- CUDA 9.0
- CUDNN V7.0
具体CUDA和CUDNN的版本要求取决于你要装的tf的版本要求,1.4至少要CUDA 8.0 + CUDNN V6.1,故此我安装的配置就是:
python 3.6 + CUDA 9.0 + CUDNN V7.0
二.配置步骤
0.检查下你的电脑是否有CUDA_Capable GP(独立显卡),打开cmd或者powershell执行如下语句:
control /name Microsoft.DeviceManager
本人电脑配的是英伟达独立显卡 GTX 1050
1.cuda9.0
(1)安装地址
https://developer.nvidia.com/cuda-90-download-archive