安装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不行,其他我也不知道。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容