win10-64安装 tensorflow gpu

安装时候出现了各种各样的问题,让人痛不欲生.搞了两天,终于大致明白了...,现在记录几个重要的点

(我的笔记本是双显卡,第二个显卡支持gpu的)

 

注意1: 版本需要对应起来,不能差太远  (如python3.7暂不支持)

          https://blog.csdn.net/omodao1/article/details/83241074

注意2: 虚拟环境不同, 命令效果不用...

//计算机环境
   C:\Users\Think>

//进入虚拟环境tensorflow(需安装)
   C:\Users\Think>activate tensorflow
   (tensorflow) C:\Users\Think>

//退出虚拟环境
   (tensorflow) C:\Users\Think>deactivate
   C:\Users\Think>

//进入虚拟环境tensorflow-gpu(需安装)
   C:\Users\Think>activate tensorflow-gpu
   (tensorflow-gpu) C:\Users\Think>

  

注意3: 忘记把cuDNN放到对应的目录,或版本不对

注意4: 网上说必须安装vs,我早就安装了.

           1,3 两个一般报错 (ImportError: DLL load failed: 找不到指定的模块。)
 

控制台下载tensorflow-gpu 命令

    (cpu版本是 tensorflow, 把下面的tensorflow-gpu替换为tensorflow就可以了)

//下载安装
pip install tensorflow-gpu 

//可以直接版本号
pip install tensorflow-gpu==11.0  

//可以直接跟地址
pip install (tensorflow-gpu的地址)


//卸载
pip uninstall

------------------------------------------------------------------------------------正式安装步骤

这里是步骤(可以根据需要自己控制版本)

先下载安装对应的版本Anaconda,cuDNN(cuDNN解压放到目录)

(cpu版本是 tensorflow, 把下面的tensorflow-gpu替换为tensorflow就可以了)

//先下载安装
C:\Users\Think> pip install tensorflow-gpu

//进入虚拟环境tensorflow-gpu
C:\Users\Think> activate tensorflow-gpu

//在虚拟环境tensorflow-gpu中安装tensorflow-gpu (两个路径都安装了就不怕了)
(tensorflow-gpu) C:\Users\Think> pip install --upgrade tensorflow-gpu

(tensorflow-gpu) C:\Users\Think> cd d:\code\python
(tensorflow-gpu) C:\Users\Think> d:

//执行代码
(tensorflow-gpu) d:\Code\python> python test.py

//安装spyder
(tensorflow-gpu) d:\Code\python> conda install spyder

//运行spyder ,第一次直接卡死,第二次运行成功,这时候就可以直接在spyder里写代码测试了
(tensorflow-gpu) d:\Code\python> spyder



//退出虚拟环境(非必须)
(tensorflow-gpu) d:\Code\python> deactivate

 

测试代码  test.py

#我这报AVX2支持的错.用这里取消掉报错
#   Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
#import os
#os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

import tensorflow as tf
hello = tf.constant('hello,tensorf')
sess = tf.Session()
print(sess.run(hello))

===============================================================

遇到的错误和解决方法

 

Could not find a version that satisfies the requirement tensorflow (from versions: )
No matching distribution found for tensorflow
错误原因:python 3.7版本尚不支持,比如安装 python3.6
(据说32位也不支持,没测试)

 

ModuleNotFoundError: No module named 'tensorflow'

这个问题很多,一般找不到的原因就是路径不对咯, 前面说在两个目录安装就是更容易找到.
我们用conda,就必须在此工作目录安装一次tensorflow,否则找不到.

打开相应的目录,如我在开始菜单打开 Spyder (Anaconda3),那么就需要在他的目录下安装tensorflow,

启动 Anaconda Powershell Prompt (Anaconda3)  命令行, pip install tensorflow 安装就可以了

如果直接在windows cmd命令窗口输入python xxx.py ,那么就在cmd 目录下安装即可,注意要设置环境变量,否则也会出现找不到.

//注: 其他错误也可能是这个问题引发的