安装GPU环境

查看cuda支持的版本
cmd中输入

cd C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi

1、tensorflow
安装cuda对照 版本
cuda10.2链接

cudnn7.6.5链接

将cudnn下的文件贴贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2对应目录下

配置环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp

dlerror: cudart64_100.dll not found最终解决办法:
https://blog.csdn.net/qq_29391809/article/details/102642314
测试代码如下:

import tensorflow as tf
print('GPU',tf.test.is_gpu_available())

版权声明:本文为qq_29507011原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。