文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

tensorflow-gpu安装100%成功(tensorflow-gpu版和tensorflow-cpu版的区别、为什么要创建虚拟环境、如何同时使用两个gpu库、tensorflow-gpu版安装)

2023-09-06 10:30

关注

1.tensorflow-gpu版和tensorflow-cpu版的区别

tensorflow-gpu版需要同时配置安装CUDA、cuDNN,而tensorflow-cpu版不需要配置,直接pip/conda install tensorflow即可安装tensorflow-cpu版本

2.为什么要创建虚拟环境

在安装gpu版本的库时通常会创建单独的虚拟环境,例如安装tensorflow-gpu,则需要利用conda create -n tensorflow python=3.7,创建一个tensorflow的虚拟环境 ,这样做的主要目的是保证tensorflow-gpu这个库不受其它库的影响,比如同时安装tensorflow-gpu与pytorch-gpu,或者同时安装不同版本的tensorflow-gpu,由于不同库或不同版本的gpu库需要配置的CUDA、cuDNN不同,导致即使都安装在base环境里,由于CUDA、cuDNN不适配,也仅能用一个gpu库,因此用到什么gpu库就单独的创建这个库的虚拟环境,这样你就不需要动之前环境中的CUDA、cuDNN,只需要在新的环境中重新配置这个新库的专属CUDA、cuDNN就OK了,只需要切换虚拟环境,这两个都能用。

3.那怎么同时用两个库呢,例如:tensorflow和opencv库

需要审视这两个库哪个对项目起主要支撑作用,是否需要更高的计算能力,比如需要tensorflow有更高的计算能力,那么就创建一个tensorflow的虚拟环境,安装tensorflow-gpu,后在这个环境中安装cpu版的opencv,即可保证两个库都能使用。
另外一种方法是,对应tensorflow-gpu库的CUDA、cuDNN的版本,找到与之匹配的opencv-gpu版本,这样即可在一个环境中用两个gpu库

4.tensorflow-gpu版安装过程

step1.安装anoconda(不要安装太高版本的)

在安装完毕后,使用Win+R调出cmd,输入conda -V,可以看到Anaconda的版本号,即为Anaconda安装成功!
具体过程可点击参考该视频

step2.对Anaconda和Python进行换源

在cmd中输入conda config --set show_channel_urls yes在C盘用户目录下会生成.condarc文件。
然后使用记事本打开,用下面的内容将其替换。

channels:  - defaultsshow_channel_urls: truechannel_alias: https://mirror.tuna.tsinghua.edu.cn/anacondadefault_channels:  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/main  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/free  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/r  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/pro  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2custom_channels:  conda-forge: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud  msys2: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud  bioconda: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud  menpo: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud  pytorch: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud  simpleitk: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud

最后保存,然后在CMD中输入conda clean -i,此时Anaconda已经换源成功!
接着是Python换源,在CMD中输入
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple即可换源成功!
参考该篇博客

step3.明确CUDA、cuDNN、Tensorflow版本的搭配

在安装Tensorflow-gpu之前,需要安装CUDA和cuDNN等。对于Tensorflow的gpu版本最重要的是CUDA、cuDNN、Tensorflow版本的搭配。
下面给出GPU版本的版本搭配:
在这里插入图片描述
具体参照该篇博客

step4.创建tensorflow虚拟环境

使用Anaconda来创建环境,这里环境名称命名为tensorflow,在CMD中输入conda create -n tensorflow python=3.7,输入y进行确认

step5.安装tensorflow-gpu(以2.1.0版本为例)

(1)先在CMD中输入activate tensorflow以激活环境
(2)输入conda install cudatoolkit=10.1,安装cudatoolkit(CUDA)(在我们的具体使用中,其实真正需要的并不是整个CUDA,而是cudatoolkit,所以我们在这里直接安装cudatoolkit,不需要再下载3个多G的CUDA来本地安装。)
(3)在激活的环境里输入conda install cudnn=7.6,安装 cuDNN
(4)在激活环境中输入pip install tensorflow-gpu==2.1.0,安装2.1.0版本 tensorflow-gpu

step6.验证

至此,Tensorflow的GPU版本已经安装成功,现在需要测试在激活环境中输入python以进入python环境。然后输入import tensorflow as tf可见如下:
在这里插入图片描述
再输入tf.test.is_gpu_available(),若最后为True表示GPU可用,证明已经成功安装Tensorflow的GPU版本
在这里插入图片描述
但是该过程可能会出现:TypeError: Descriptors cannot not be created directly. If this call came from a _pb2.py file……这个问题,如果出现采用Step7.

step7.解决TypeError

(1)打开CMD输入activate tensorflow进入激活环境
(2)在激活环境中输入pip uninstall protobuf
(3)继续输入pip install protobuf==3.20.1
即可完美解决问题,重新测试,即可成功!

来源地址:https://blog.csdn.net/sunshine_boy1/article/details/129923839

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯