之前都是使用tensorflow的cpu版本,没注意自己这台老人机居然有GPU。便决定试一试看看自己电脑能不能跑起GPU。(各种百度谷歌)记录如下:
首先:检查自己的GPU是否可以使用:
CUDA:https://developer.nvidia.com/cuda-gpus
居然找到自己的GPU..哈哈哈(意外)
接下来:需要安装CUDA + cuDNN。
刚开始无脑:使用CUDA10.0 +cuDNN 7 .4.2+tensorflow1.12发现报错如下(没截图):
File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File