tensorflow-gpu 2.6版本安装及cuda、cudnn配置

tensorflow-gpu 2.6 win10

  • 查看对应版本
  • 下载包
  • 安装cuda
  • 安装cudnn
  • 测试代码

查看对应版本

官网查看
tensorflow-gpu 2.6版本安装及cuda、cudnn配置_第1张图片

下载包

cuda各版本下载地址
cudnn各版本下载地址

tensorflow下载代码

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.6.0

安装cuda

参考网址

安装cudnn

把对应目录解压到cuda文件夹下即可,不要按照上面的网址安装!

测试代码

返回True即可,警告可忽略。

import tensorflow as tf
tf.test.is_gpu_available()

你可能感兴趣的:(tensorflow,python)