安装tensorflow-gpu

2018-04-22  本文已影响0人  tongues

下载CUDA 9.0安装
下载cudnn 7.0,解压缩,将解压缩路径拷贝到环境变量path中(安装目录和bin文件目录)
pip install tensorflow-gpu==1.8.0
搞定!!!!!!!
注:1.7的tensorflow版本只能使用9.0的CUDA。9.1不行,其他我也不知道。

上一篇下一篇

猜你喜欢

热点阅读