美文网首页
MacOS 安装Tensorflow,带GPU加速的那种

MacOS 安装Tensorflow,带GPU加速的那种

作者: 青云直上 | 来源:发表于2022-05-03 14:40 被阅读0次

    苹果没有nvidia的gpu,要充分利用自身的GPU资源发挥TensorFlow的多线程处理能力,可以使用苹果自带的metal包。

    官方网站

    系统要求

    最好是OSX 12.xx以上,自带ML Compute,否则要自己安装

    以下实例为intel CPU, AMD GPU及问题解决。 M1版本以上看官网。

    创建环境

    conda create --name tf python=3.8
    #注意目前官方支持3.8
    

    激活环境

    source activate tf 
    

    安装库

    pip install tensorflow-macos
    pip install tensorflow-metal
    #注意要使用pip,conda找不到
    

    问题来了,显示找不到相对应的版本的包,我可是新建的环境啊!!!

    网上找了一大圈,有朋友找到说是MacOS Big Sur 导致的版本混乱。 Python无法识别正确的macos版本。

    解决办法:

    找到虚拟环境下pip对系统版本判定的function,手动指定macos 版本。

    比如在

    ~/miniconda3/envs/tf/lib/python3.8/site-packages/pip/_vendor/packaging/tags.py

    搜索platform.mac_ver()
    找到

     version_str, _, cpu_arch = platform.mac_ver()
    

    在后面加一行

    version_str = "12.3"  #根据你自己的macos填
    

    再运行

    pip install tensorflow-macos
    pip install tensorflow-metal
    

    问题解决!

    相关文章

      网友评论

          本文标题:MacOS 安装Tensorflow,带GPU加速的那种

          本文链接:https://www.haomeiwen.com/subject/btobyrtx.html