此处只讲配置tensorflow-GPU(1.13.0)+keras( 2.2.4)部分:
本人电脑GPU型号:NVIDA GeForce GTX 1050 Cuda支持:10.0.132
1)tensorflow-gpu版本与CUDA版本对应关系参考链接:
https://blog.youkuaiyun.com/qq_27825451/article/details/89082978?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1
对应需要安装tensorflow_gpu-1.13.0 cuDNN7.4+ CUDA10.0
2)Win10安装CUDA10和cuDNN+配置参考:
https://blog.youkuaiyun.com/qq_37296487/article/details/83028394?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1
cuDNN复制到CUDA对应文件中:copy------》past
按照自己的情况来,图中是我的解压和安装路径:(注意是路径中的文件)
安装版本号:
cudnn-10.0-windows10-x64-v7.6.4.38.zip
cuda_10.0.130_win10_network.exe
百度云下载链接:https://pan.baidu.com/s/1pEvxuMfpn2TpJM0Uqnld1w
提取码:i287
3)配置tensorflow_gpu-1.13.0
创建一个虚拟环境.
conda create -n tensorflow_gpu python=3.5
激活环境 在命令行中输入:
conda activate tensorflow_gpu
安装tensorflow_gpu-1.13.1
pip install tensorflow-gpu==1.13.1 –i https://pypi.tuna.tsinghua.edu.cn/simple
4)安装tensorflow_gpu版本的keras(’ 2.2.4’)
pip install keras==2.2.4 –i https://pypi.tuna.tsinghua.edu.cn/simple
配置附属插件例如:matplotlib、cv2、PIL等
pip install matplotlib
pip install opencv-python
pip install pillow