1.安装gpu之前一定要卸载到cpu版本的,否则默认是不会使用gpu版的;
第0步:卸载protobuf
pip uninstall protobuf
第1步:卸载tensorflow
pip uninstall tensorflow pip uninstall tensorflow-gpu
第2步:强制重新安装支持GPU的Tensorflow
pip install --upgrade --force-reinstall tensorflow-gpu
2.tensorflow的版本一定要cuda和cudnn对应
tensorflow-gpu==1.3.0
cuda=8.0
cudnn=6.0
3.先安装cuda和cudnn,配置相应的环境变量,再安装tensorflow-gpu
cuda 还需要在环境变量里的path全局变量里加入,这个下面的bin和lib\x64目录的路径。
cudnn 下载下来后将压缩包里各文件夹里的文件放到对应目录下就好
4.安装tensorflow-gpu
pip install --upgrade --force-reinstall D:\data\tensorflow_gpu-1.3.0-cp35-cp35m-win_amd64.whl
5.安装完毕之后,请把cmd小窗关闭之后,再打开看看,import tensorflow,不报错即可
6.其他问题:
Cannot uninstall 'html5lib'. It is a distutils installed project and thus we cannot accurately....
Cannot uninstall 'html5lib'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.
问题解决:请找到python下面的路径
然后再找到这个东东把它删除就可以了
类似的,以后这种问题的解决方式都可以这样入手。
CPU和GPU对比来看速度提升了3-4倍左右
Inv3和vgg16比起来速度快了1倍,但是准确率低了10%,也不排除因为vgg16多加了一层卷积的原因
CPU版运行速度——inception_v3:
Found 2110 images belonging to 4 classes.
Found 120 images belonging to 4 classes.
Epoch 1/50
1/30 [>.............................] - ETA: 2:04 - loss: 1.1210 - acc: 1.0000
2/30 [=>............................] - ETA: 1:26 - loss: 1.7921 - acc: 0.5000
3/30 [==>...........................] - ETA: 1:12 - loss: 1.5732 - acc: 0.3333
4/30 [===>..........................] - ETA: 1:04 - loss: 1.7981 - acc: 0.2500
5/30 [====>.........................] - ETA: 58s - loss: 1.9207 - acc: 0.2000
6/30 [=====>........................] - ETA: 54s - loss: 1.9304 - acc: 0.1667
7/30 [======>.......................] - ETA: 51s - loss: 1.8344 - acc: 0.1429
8/30 [=======>......................] - ETA: 49s - loss: 1.7362 - acc: 0.2500
9/30 [========>.....................] - ETA: 46s - loss: 1.6440 - acc: 0.3333
10/30 [=========>....................] - ETA: 43s - loss: 1.6094 - acc: 0.3000
11/30 [==========>...................] - ETA: 41s - loss: 1.6335 - acc: 0.2727
12/30 [===========>..................] - ETA: 38s - loss: 1.5960 - acc: 0.2500
13/30 [============>.................] - ETA: 36s - loss: 1.5524 - acc: 0.3077
14/30 [=============>................] - ETA: 33s - loss: 1.5241 - acc: 0.2857
15/30 [==============>...............] - ETA: 31s - loss: 1.4929 - acc: 0.2667
16/30 [===============>..............] - ETA: 29s - loss: 1.5141 - acc: 0.2500
17/30 [================>.............] - ETA: 27s - loss: 1.5264 - acc: 0.2353
18/30 [=================>............] - ETA: 24s - loss: 1.4985 - acc: 0.2778
19/30 [==================>...........] - ETA: 22s - loss: 1.4797 - acc: 0.2632
20/30 [===================>..........] - ETA: 20s - loss: 1.4550 - acc: 0.3000
21/30 [====================>.........] - ETA: 18s - loss: 1.4931 - acc: 0.2857
22/30 [=====================>........] - ETA: 16s - loss: 1.5223 - acc: 0.2727
23/30 [======================>.......] - ETA: 14s - loss: 1.5446 - acc: 0.2609
24/30 [=======================>......] - ETA: 12s - loss: 1.5493 - acc: 0.2500
25/30 [========================>.....] - ETA: 10s - loss: 1.5293 - acc: 0.2800
26/30 [=========================>....] - ETA: 8s - loss: 1.5088 - acc: 0.3077
27/30 [==========================>...] - ETA: 6s - loss: 1.5268 - acc: 0.2963
28/30 [===========================>..] - ETA: 4s - loss: 1.5075 - acc: 0.3214
29/30 [============================>.] - ETA: 2s - loss: 1.5217 - acc: 0.3103
2110/30 [==================================================================]
- 4037s 2s/step - loss: 1.3876 - acc: 0.2864 - val_loss: 1.3181 - val_acc: 0.3833
GPU版运行速度——inception_v3:
name: GeForce 920MX
major: 5 minor: 0 memoryClockRate (GHz) 0.993
pciBusID 0000:01:00.0
Total memory: 2.00GiB
Free memory: 1.66GiB
2018-06-27 14:38:27.880300: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:976] DMA: 0
2018-06-27 14:38:27.880463: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:986] 0: Y
2018-06-27 14:38:27.880646: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1045] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce 920MX, pci bus id: 0000:01:00.0)
Found 2110 images belonging to 4 classes.
Found 120 images belonging to 4 classes.
Epoch 1/50
1/30 [>.............................] - ETA: 4:07 - loss: 1.4332 - acc: 0.0000e+00
2/30 [=>............................] - ETA: 2:09 - loss: 1.7041 - acc: 0.0000e+00
3/30 [==>...........................] - ETA: 1:29 - loss: 1.5663 - acc: 0.0000e+00
4/30 [===>..........................] - ETA: 1:08 - loss: 1.6676 - acc: 0.0000e+00
5/30 [====>.........................] - ETA: 56s - loss: 1.5959 - acc: 0.0000e+00
6/30 [=====>........................] - ETA: 47s - loss: 1.5104 - acc: 0.1667
7/30 [======>.......................] - ETA: 41s - loss: 1.4269 - acc: 0.2857
8/30 [=======>......................] - ETA: 36s - loss: 1.5028 - acc: 0.2500
9/30 [========>.....................] - ETA: 32s - loss: 1.5329 - acc: 0.2222
10/30 [=========>....................] - ETA: 29s - loss: 1.5658 - acc: 0.2000
11/30 [==========>...................] - ETA: 26s - loss: 1.5766 - acc: 0.1818
12/30 [===========>..................] - ETA: 23s - loss: 1.5699 - acc: 0.1667
13/30 [============>.................] - ETA: 21s - loss: 1.5321 - acc: 0.2308
14/30 [=============>................] - ETA: 19s - loss: 1.5368 - acc: 0.2143
15/30 [==============>...............] - ETA: 17s - loss: 1.5492 - acc: 0.2000
16/30 [===============>..............] - ETA: 16s - loss: 1.5475 - acc: 0.1875
17/30 [================>.............] - ETA: 14s - loss: 1.5400 - acc: 0.1765
18/30 [=================>............] - ETA: 13s - loss: 1.5183 - acc: 0.2222
19/30 [==================>...........] - ETA: 11s - loss: 1.5200 - acc: 0.2105
20/30 [===================>..........] - ETA: 10s - loss: 1.5127 - acc: 0.2000
21/30 [====================>.........] - ETA: 9s - loss: 1.4956 - acc: 0.2381
22/30 [=====================>........] - ETA: 8s - loss: 1.5051 - acc: 0.2273
23/30 [======================>.......] - ETA: 7s - loss: 1.4882 - acc: 0.2609
24/30 [=======================>......] - ETA: 5s - loss: 1.4699 - acc: 0.2917
25/30 [========================>.....] - ETA: 4s - loss: 1.4511 - acc: 0.3200
26/30 [=========================>....] - ETA: 3s - loss: 1.4623 - acc: 0.3077
27/30 [==========================>...] - ETA: 2s - loss: 1.4683 - acc: 0.2963
28/30 [===========================>..] - ETA: 1s - loss: 1.4519 - acc: 0.3214
29/30 [============================>.] - ETA: 0s - loss: 1.4527 - acc: 0.3103
2110/30 [================================================================]
- 1443s 684ms/step - loss: 1.3879 - acc: 0.2864 - val_loss: 1.3937 - val_acc: 0.2500
GPU版运行速度——vgg16:
name: GeForce 920MX
major: 5 minor: 0 memoryClockRate (GHz) 0.993
pciBusID 0000:01:00.0
Total memory: 2.00GiB
Free memory: 1.66GiB
Found 2110 images belonging to 4 classes.
Found 120 images belonging to 4 classes.
Epoch 1/50
1/30 [>.............................] - ETA: 3:11 - loss: 1.0740 - acc: 0.0000e+00
2/30 [=>............................] - ETA: 1:45 - loss: 2.3876 - acc: 0.0000e+00
3/30 [==>...........................] - ETA: 1:16 - loss: 1.7835 - acc: 0.3333
4/30 [===>..........................] - ETA: 1:01 - loss: 2.0477 - acc: 0.2500
5/30 [====>.........................] - ETA: 51s - loss: 2.1607 - acc: 0.2000
6/30 [=====>........................] - ETA: 45s - loss: 2.1200 - acc: 0.1667
7/30 [======>.......................] - ETA: 40s - loss: 1.9350 - acc: 0.2857
8/30 [=======>......................] - ETA: 36s - loss: 1.9065 - acc: 0.2500
9/30 [========>.....................] - ETA: 33s - loss: 1.7886 - acc: 0.3333
10/30 [=========>....................] - ETA: 30s - loss: 1.7543 - acc: 0.3000
11/30 [==========>...................] - ETA: 27s - loss: 1.7086 - acc: 0.2727
12/30 [===========>..................] - ETA: 25s - loss: 1.6439 - acc: 0.3333
13/30 [============>.................] - ETA: 23s - loss: 1.6839 - acc: 0.3077
14/30 [=============>................] - ETA: 21s - loss: 1.6510 - acc: 0.2857
15/30 [==============>...............] - ETA: 19s - loss: 1.6018 - acc: 0.3333
16/30 [===============>..............] - ETA: 18s - loss: 1.5791 - acc: 0.3125
17/30 [================>.............] - ETA: 16s - loss: 1.6078 - acc: 0.2941
18/30 [=================>............] - ETA: 15s - loss: 1.5725 - acc: 0.3333
19/30 [==================>...........] - ETA: 13s - loss: 1.5855 - acc: 0.3158
20/30 [===================>..........] - ETA: 12s - loss: 1.5892 - acc: 0.3000
21/30 [====================>.........] - ETA: 10s - loss: 1.5755 - acc: 0.2857
22/30 [=====================>........] - ETA: 9s - loss: 1.5544 - acc: 0.2727
23/30 [======================>.......] - ETA: 8s - loss: 1.5273 - acc: 0.3043
24/30 [=======================>......] - ETA: 7s - loss: 1.5337 - acc: 0.2917
25/30 [========================>.....] - ETA: 5s - loss: 1.5565 - acc: 0.2800
26/30 [=========================>....] - ETA: 4s - loss: 1.5350 - acc: 0.3077
27/30 [==========================>...] - ETA: 3s - loss: 1.5138 - acc: 0.3333
28/30 [===========================>..] - ETA: 2s - loss: 1.4927 - acc: 0.3571
29/30 [============================>.] - ETA: 1s - loss: 1.4863 - acc: 0.3448
2110/30 [=======================================================]
- 2033s 964ms/step - loss: 1.3345 - acc: 0.3815 - val_loss: 1.3081 - val_acc: 0.4333
Epoch 2/50