博客

1、安装cuda(查看电脑cuda版本，打开nvidia控制面板-系统信息-组件，见NACUDA.DLL的产品信息)

下载网址

https://developer.nvidia.com/cuda-10.2-download-archive

安装时选择自定义安装，只安装cuda模块即可。

2、下载对应版本的cudnn

下载地址：

https://developer.nvidia.com/rdp/cudnn-download

配置环境变量参考

https://blog.csdn.net/qq_37296487/article/details/83028394

3、安装Visual C++（选择对应版本）安装后需要重启

https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads

4、安装tensorflow-gpu

从豆瓣源安装速度快一些

pip install tensorflow-gpu -i https://pypi.doubanio.com/simple

5、进入python

import tensorflow as tf

若不报错，则安装成功，若出现Could not load dynamic library ‘cudart64_101.dll’; dlerror:

可以通过从网上下载cudart64_101.dll文件，也可以通过先安装cuda10.1,再安装更高版本，将10.1中cudart64_101.dll拷贝到对应文件夹中。

6、测试是否在GPU上运行

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

import tensorflow as tf
tf.test.gpu_device_name()