按照tensorflow教程https://www.tensorflow.org/install/install_windows安装GPU版本tensorflow,
pip3 install --upgrade tensorflow-gpu
从 shell 中调用 Python,如下所示:
$ python
在 Python 交互式 shell 中输入代码:
>>> import tensorflow as tf
测试时报Could not find ‘cudart64_90.dll错误
查询一番才发现是版本问题,默认下载的是CUDA 9.2,但tensorflow与CUDA不搭配,重新下载CUDA 9.0并安装后,
(记得重新开一个cmd窗口,以使环境变量生效)
再次执行又报错:
ImportError: Could not find ‘cudnn64_7.dll’. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs. You may install the necessary DLL by downloading cuDNN 7 from this URL: https://developer.nvidia.com/cudnn
也是cuDNN版本不对的问题,
重新下载Download cuDNN v7.2.1 (August 7, 2018), for CUDA 9.0,然后解压对应的文件到CUDA的安装目录下,如图
就可以正常导入了
>>> import tensorflow as tf