本文主要是介绍tensorflow-gpu 安装 tensorflow1.12 + CUDA 9.0 + cudnn 7.5 同时安装gpu版本与cpu版本,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
现在 windows10 支持 pip install tensorflow-gpu 直接安装
一、安装tensorflow-gpu
但是需要注意:
1、电脑要支持 CUDA(检查自己的GPU看支持到哪个版本,GTX960M支持9.0)
2、需要下载cudnn,这个深度学习加速库,并把cudnn下面每一项放到CUDA对应目录
可能出现的问题:
1、找不到DLL的情况
这个问题可能是,你没有把cudnn对应文件正确的放入CUDA/v9.0/xx目录下面。
也可能是你的CUDA与cudnn版本不兼容
还有可能就是你所pip安装的tensorflow和你安装的CUDA版本不兼容(因为pip默认安装当下最新tensorflow,然而他并不一点和你所安装的CUDA版本想兼容,例如我之前安装的tensorflow1.13,他和我安装的CUDA9.0不兼容,所以需要安装tensorflow1.12才行)
2、至于出现了一些警告:
可以不用管,也可以通过忽略解决。
二、安装tensorflow -cpu
1、需要另外设置一个虚拟环境进行安装,同时尽量保持 python版本一致,tensorflow版本一致
2、pip安装方式默认安装的是manylinux1_X86这是一个通用版本,在win10上面会有警告。
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
可直接到http://mirrors.aliyun.com/pypi/simple/tensorflow/下载amd_64版本whl文件安装。
3、若安装whl文件时候报错zipfile.BadZipFile: File is not a zip file,则说明安装包下载过程出错,重新下载即可。
查看GPU是否支持CUDA(计算能力大于3.1即可): https://developer.nvidia.com/cuda-gpus
tensorflow与CUDA版本对应以及镜像源:https://blog.csdn.net/yuejisuo1948/article/details/81043962
安装步骤: https://www.jianshu.com/p/aafd88033867
这篇关于tensorflow-gpu 安装 tensorflow1.12 + CUDA 9.0 + cudnn 7.5 同时安装gpu版本与cpu版本的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!