前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >tensorflow-gpu版本安装

tensorflow-gpu版本安装

作者头像
sofu456
发布2021-12-06 14:17:54
1.2K0
发布2021-12-06 14:17:54
举报
文章被收录于专栏:sofu456sofu456

is_gpu_available

代码语言:javascript
复制
import tensorflow as tf
tf.test.is_gpu_available()    //返回false,gpu不成功

安装cuda,cudnn 地址:https://developer.nvidia.com/cuda-toolkit-archive

tensorflow各版本对应的cuda查看链接:https://tensorflow.google.cn/install/source#linux

我安装的tensorflow2.4,cuda 11,cudnn 8安装完成后,提示 Could not load dynamic library ‘cusolver64_10.dll’ 解决办法:cusolver64_11.dll复制一份,改名为cusolver64_10.dll

如果还不行: 更新nvdia驱动,设置显卡为geforce显卡 https://blog.csdn.net/flana/article/details/104772614

出现以下提示为正常

在这里插入图片描述
在这里插入图片描述
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2021-05-11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • is_gpu_available
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档