tensorlflow 맥os(macOS) 에서 gpu 사용 설정 하기

반응형
    728x90
    반응형

    Macos 유저 tensorflow gpu 사용 설정 하는 법

    1. Homebrew 설치 (이미 설치된 경우 생략):

    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

    2. 필요한 종속성 설치 (이미 설치된 경우 생략):

    brew install python
    brew install cmake
    brew install libjpeg-turbo
    

    3. miniforge 설치(이미 설치된 경우 생략):

    curl -L -O https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-MacOSX-arm64.sh
    bash Miniforge3-MacOSX-arm64.sh
    

     4. 사용중인 가상환경 진입(가상 환경 없으면 만들어주기):

    conda create -n tf-metal python=3.9
    conda activate tf-metal
    

     5. tensorflow-macos 및 tensorflow-metal 플러그인 설치:

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

    6. TensorFlow 테스트:

    import tensorflow as tf
    print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
    728x90
    반응형

    댓글

    Designed by JB FACTORY