我的版本配置:
*Python3.8
*CUDA10.2
**cudnn:只要是适合CUDA10.2的版本都可以(但这里有个小问题,有些版本会出现not opened dynamic library cudart64_101.dll,就是找不到文件,有些是101,有些是102,解决办法:下载对应的cudart64_101.dll添加到文件夹下即可,可能不会马上起作用,这时就需要重启电脑了)
*tensorflow2.3.0
*tensorflow-gpu2.3.0
1、首先电脑要支持GPU(英伟达);
2、看一下显卡的CUDA版本号,然后下载对应的cuda(需要安装)和cudnn,将cudnn中的文件复制到cuda对应文件夹下;
3、下面是简单一段查看是否成功调用GPU
import tensorflow as tf
import os
from tensorflow.python.client import device_lib
print(devic