首选推荐安装Mambaforge, 也可以安装miniforge3。
项目链接:https://github.com/conda-forge/miniforge/
conda create -n tf270 -c apple tensorflow-deps conda activate tf270 pip install tensorflow-macos tensorflow-metal测试安装是否成功
import tensorflow as tf print(tf.__version__) print(tf.version.VERSION) from tensorflow.python.client import device_lib print(device_lib.list_local_devices())
结果如下GPU即为安装成功:
(tf270) hzw@bogon ~ % python Python 3.9.7 | packaged by conda-forge | (default, Sep 29 2021, 19:24:02) [Clang 11.1.0 ] on darwin Type "help", "copyright", "credits" or "license" for more information. >>> from tensorflow.python.client import device_lib >>> print(device_lib.list_local_devices()) metal device set to: Apple M1 systemMemory: 8.00 GB maxCacheSize: 2.67 GB 2021-12-15 16:38:14.922342: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:305] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support. 2021-12-15 16:38:14.923134: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:271] Created TensorFlow device (/device:GPU:0 with 0 MB memory) -> physical PluggableDevice (device: 0, name: metaL, pci bus id:参考链接) [name: "/device:CPU:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 16585735293412376744 xla_global_id: -1 , name: "/device:GPU:0" device_type: "GPU" locality { bus_id: 1 } incarnation: 17634086545751551048 physical_device_desc: "device: 0, name: metaL, pci bus id: " xla_global_id: -1 ] >>>
https://developer.apple.com/metal/tensorflow-plugin/
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)