用Pycharm跑VGG16报了上面的警告,还提示“Skipping registering GPU devices…”,意思是你的代码可以运行,但不是用GPU跑的,而是CPU!心痛啊!明明已经配置了tensorflow、cuda等等,看来还是没有配置好,那就继续配置吧!
发现问题
还是自己的tensorflow版本和cuda版本不对应导致的,查看电脑的显卡驱动版本和cuda版本
解决问题
重装tensorflow!
1 打开控制面板,点击“硬件和声音”
2 点击“DVIDIA控制面板”
3 点击“系统信息”
4 在这里可以看到驱动的版本是376.54,点击“组件”
5 就可以看到我们需要的cuda版本,我的是CUDA 8.0.0(好老的版本啊…)(cuda已安装好了)
6 好!找到这两个关键数据之后,我们开始找对应的tensorflow版本,我的是tensorflow_gpu-1.4.0。因为我的cuda版本是正确的,所以我只需要重装tensorflow即可,如果你们的cuda版本也不对,就先安装cuda之后再安装tensorflow!!
7 打开anaconda prompt
conda create -n tensorflow-gpu pip python=3.6 #创建tensorflow-gpu的虚拟环境
执行完后,在anaconda navigator中会出现tensorflow-gpu的虚拟环境
8 anaconda prompt中
activate tensorflow-gpu
pip install tensorflow-gpu==1.4.0
9 8完成之后输入pip list,会出现如下信息
pip list
说明安装成功!
10 进入pycharm进行配置,步骤如下
11 运行pycharm中的代码,木的问题了呀!
第一次写这么长的博文,有写的不好的地方请多多关照!希望能够帮到你哦!