GPU环境配置与Faster-Rcnn训练记录
我是参考Bubbliiiing博主的内容进行学习的,这是他的主页:
https://blog.youkuaiyun.com/weixin_44791964?spm=1001.2014.3001.5509
关于windows下的tensorflow-gpu=1.13.2环境配置可参考:
https://blog.youkuaiyun.com/weixin_44791964/article/details/104702142
我在这里只是记录一下我自己的操作流程。
具体的环境内容如下:
tensorflow-gpu:1.13.2
keras:2.1.5
numpy:1.17.4
我的电脑配置:
win10, 1050ti, 显存4G
Anaconda的安装就不说了,直接从cuda的安装说起,安装时需要注意以下几个问题:
- 安装cuda时,许可协议里选自定义,只选择CUDA,并且取消勾选CUDA里的visual studio integration选项。
- 添加系统环境变量:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64,并上移到第3位。(这一步很重要)
查看cuda的版本
tensorflow-gpu的环境创建
<