Tensorflow下载

查阅Tensorflow与python cuda cudnn版本对应问题

https://tensorflow.google.cn/install/source_windows
windows:10
python:3.8
tensorflow:gpu2.3.0

cuda10.1下载

链接: https://pan.baidu.com/s/19mI0BeWLtXmtB53RJCXn2A 提取码: n37w

cudnn在官网注册后选择对应版本(在tensorflow官网查到的版本)下载

检测是否成功

import tensorflow as tf
gpu_ok=tf.test.is_gpu_available()
print(gpu_ok)

为true表示成功

posted @ 2020-11-03 22:31  小帆敲代码  阅读(273)  评论(0编辑  收藏  举报