tensorflow-gpu安装的一些注意

Posted 2021-01-16 robird

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

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

1.1，安装9.0 patch1之前，先删除掉C:Program FilesNVIDIA Corporationinstaller2，否则闪退。

1.2，安装cuDnn后，检查下%PATH%里是否有C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0in和libnvvp。

　　以及%CUDA_PATH%=C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0

　　以及%CUDA_PATH_V9_0%=C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.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));