现在 windows10 支持 pip install tensorflow-gpu 直接安装
一、安装tensorflow-gpu
但是需要注意:
1、电脑要支持 CUDA(检查自己的GPU看支持到哪个版本,GTX960M支持9.0)
2、需要下载cudnn,这个深度学习加速库,并把cudnn下面每一项放到CUDA对应目录
可能出现的问题:
1、找不到DLL的情况
这个问题可能是,你没有把cudnn对应文件正确的放入CUDA/v9.0/xx目录下面。
也可能是你的CUDA与cudnn版本不兼容
还有可能就是你所pip安装的tensorflow和你安装的CUDA版本不兼容(因为pip默认安装当下最新tensorflow,然而他并不一点和你所安装的CUDA版本想兼容,例如我之前安装的tensorflow1.13,他和我安装的CUDA9.0不兼容,所以需要安装tensorflow1.12才行)
2、至于出现了一些警告:
可以不用管,也可以通过忽略解决。
二、安装tensorflow -cpu
1、需要另外设置一个虚拟环境进行安装,同时尽量保持 python版本一致,tensorflow版本一致
2、pip安装方式默认安装的是manylinux1_X86这是一个通用版本,在win10上面会有警告。
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
可直接到http://mirrors.aliyun.com/pypi/simple/tensorflow/下载amd_64版本whl文件安装。
<