安装tensorflow-gpu

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

你可能感兴趣的:(安装tensorflow-gpu)