首页 > 其他 > 详细

tensorflow-gpu安装的一些注意

时间:2018-10-31 15:18:52      阅读:158      评论:0      收藏:0      [点我收藏+]

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

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安装的一些注意

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!