踩坑总结：安装使用TensorFlow-gpu

• 要有支持的GPU显卡，对应驱动
• 安装Python3.6

 

• 推荐conda（非最新版强迫症患者）：

Anaconda3-5.2.0 Prompt命令行中：conda install -c anaconda tensorflow-gpu，会顺带把Cuda和CuDNN也给装了。

检查gpu是否可用，列出所有可用设备：

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

使用gpu：

import tensorflow as tf

with tf.device("/gpu:0"):
    # Setup operations

with tf.Session() as sess:
    # Run your code

 

• pip原生安装：注意版本对应

安装cuda9.0，配置环境变量bin，验证nvcc -V

安装cudnn7.0（解压到cuda根目录），配置环境变量bin

安装TensorFlow-gpu，win+R,cmd系统命令行pip3 install --upgrade tensorflow-gpu

验证import tensorflow

 

• 注意：有些操作无法用GPU完成，只能换回CPU。

能用conda就用conda，conda自动安装依赖，方便。

 

• 参考：

https://www.tensorflow.org/install/install_windows

https://blog.csdn.net/weixin_39290638/article/details/80045236

https://learningtensorflow.com/lesson10/

https://stackoverflow.com/questions/43577923/cannot-import-tensorflow-for-gpu-on-windows-10新版重命名为旧版

 

• 更多：
• moviepy的使用：

需要前置ffmpeg，留意错误信息。

moviepy与OpenCV冲突，只能装低版本opencv3.2.0；

在pip安装moviepy 0.2.3.2，在conda中没有；

在pip安装jupyter。

win32错误，可以换为Ubuntu