tips (8) mac (5) m1 (1) tensorflow (1) gpu (1)

[tips] m1 macbook tensorflow gpu 설치

 conda install -c apple tensorflow-deps
 python -m pip install tensorflow-macos
 python -m pip install tensorflow-metal

import tensorflow as tf
tf.__version__
tf.config.list_physical_devices()

"""
[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU'),
 PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
"""

graph error 발생시

python -m pip install tensorflow-macos==2.9
python -m pip install tensorflow-metal==0.5.0

참고
[1] https://developer.apple.com/forums/thread/721619

이 문서는 jhy156456에 의해 작성되었습니다.
마지막 수정 날짜:2022-12-21 13:00:00