您现在的位置是:首页 > 技术教程 正文

【精简】2023年最新Windows安装GPU版本的tensorflow(含bug记录及解决)

admin 阅读: 2024-03-24
后台-插件-广告管理-内容页头部广告(手机)

GPU版(2023)的Tensorflow安装(GTX1060)

在这里插入图片描述

一、Anaconda虚拟环境的创建

1、cmd进入命令行,输入:

conda create -n py38 python=3.8
  • 1

-n:自定义的虚拟环境名,我的虚拟环境为py38;
后选定python版本,选择python 3.8;

二、检查cuda与cudnn版本号

1、激活已创建好的虚拟环境,命令行输入:

conda activate py38
  • 1

2、命令一:用于检查当前的cuda版本号

conda search cuda
  • 1

在这里插入图片描述

3、命令二:用于检查当前的cudnn版本号

conda search cudnn
  • 1

在这里插入图片描述

三、安装对应的cuda与cudnn版本,不用预装cuda和cudnn

1、选择最终的cuda和cudnn搭配组合:

conda install cudatoolkit=10.1.243 conda install cudnn=7.6.5 pip install tensorflow-gpu==2.2.0
  • 1
  • 2
  • 3

同时,GPU对应版本查询网址:
Tensorflow官网
在这里插入图片描述
2、问题解决
2.1、问题描述一

按照上文提到的命令依次安装,出现 TypeError: Descriptors cannot not be created directly(局部错误)
  • 1
  • 2

解决方法(成功):

pip install protobuf==3.19.0
  • 1

2.2、问题描述二(成功

Could not load dynamic library cudart64_101.dll(局部)
  • 1

解决方法:

下载cudart64_101.dll文件
  • 1

文件资源网址:
cudart64_101.dll
3、最终成果检测命令及结果展示

import tensorflow as tf print(tf.__version__) tf.test.is_gpu_available()
  • 1
  • 2
  • 3
  • 4

结果展示:
在这里插入图片描述
在这里插入图片描述

四、有问题,随时私信留言,看到后回,求点赞收藏和关注

标签:
声明

1.本站遵循行业规范,任何转载的稿件都会明确标注作者和来源;2.本站的原创文章,请转载时务必注明文章作者和来源,不尊重原创的行为我们将追究责任;3.作者投稿可能会经我们编辑修改或补充。

在线投稿:投稿 站长QQ:1888636

后台-插件-广告管理-内容页尾部广告(手机)
关注我们

扫一扫关注我们,了解最新精彩内容

搜索
排行榜