Mask_RCNN入门（三）—— 使用gpu进行训练和测试

cpu换成gpu

之前的工作：
Mask_RCNN入门（一）—— 跑通代码
Mask_RCNN入门（二）—— 训练自己的数据集
其实是在CPU上跑的，这一点可以在requirements.txt中可以看到：

tensorflow就是CPU版，GPU版本是tensorflow-gpu。
所以第一步是卸载tensorflow，安装tensorflow-gpu，为了避免兼容性冲突，tensorflow-gpu同样使用1.6.0版本。
卸载tensorflow：

python -m pip uninstall tensorflow

安装tensorflow-gpu：

python -m pip install tensorflow-gpu==1.6.0

driver，CUDA，cuDNN和tensorflow-gpu

安装完tensorflow-gpu之后，如果直接运行demo.py或者train.py会报错：ImportError: Could not find 'cudart64_90.dll’

其实信息明确了，到指定网址下载安装CUDA 9.0。
这里开一下上帝视角：CUDA，包括后面同样会报错需要下载的cuDNN，它们是什么，还有和CUDA版本紧密相关的显卡驱动又是什么？