安装搭配anacoda+tensorflow-gpu+pycharm

电脑 Windows10+ NVIDIA GEFORCE GTX 1050 安装时出了很多问题,看了很多帖子,

https://blog.csdn.net/lichuan0987/article/details/80703935

https://blog.csdn.net/gangeqian2/article/details/79358543

Anaconda 创建虚拟环境 用的5.1.0版本for python 3.6  记得添加到PATH

创建虚拟环境  conda create --name tensorflow-gpu python=3.6

虚拟环境名称就是 tensorflow-gpu

安装上了 python 3.6

activate tensorflow-gpu 启动虚拟环境

CUDA 9.0

cudnn 7.1.3 把解压文件复制粘贴到 CUDA安装的相应文件夹

在虚拟环境中 安装 pip install tensorflow-gpu==1.8.0

安装成功,个人没做其他的尝试,怕出更多的问题,pycharm >>setting>>选择到虚拟环境中tensorflow-gpu中的python3.6.exe

运行时会报Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 错误

不影响使用,不用管。就是没发挥CPU性能.

同一个案例运行从9000多秒变成了30多微秒。

猜你喜欢

转载自blog.csdn.net/qq_41109499/article/details/85126802