tensorflow-gpu安装的一些注意

标签：win   检查   ram   conf   .config   python   其他   \n   dia   

按正确的顺序安装，严格安装特定的版本

1，下载和安装严格版本的cuda和cuDnn,其他版本的不干活。比如要求9.0你就不能装9.1。https://www.tensorflow.org/install/install_windows

1.1，安装9.0 patch1之前，先删除掉C:\Program Files\NVIDIA Corporation\installer2，否则闪退。

1.2，安装cuDnn后，检查下%PATH%里是否有C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin和libnvvp。

　　以及%CUDA_PATH%=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

　　以及%CUDA_PATH_V9_0%=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

2，安装显卡驱动，这个可以装最新的。

3，pip install tensorflow-gpu。注意不要安装非“-gpu”的CPU版。如果安了，俩全卸载，再单安-gpu。

3，pip uninstall protobuf, pip install protobuf==3.6.0(严格和你的python版本相同，3.6.1就不干活)

4，进python验证一下，import tensorflow as tf; sess=tf.Session(config=tf.ConfigProto(log_device_placement=True));

tensorflow-gpu安装的一些注意

标签：win   检查   ram   conf   .config   python   其他   \n   dia   

原文地址：https://www.cnblogs.com/robird/p/9882857.html