Window 10에서 Tensorflow GPU 환경 구성하기

Machine Learning

engyjoon 2021. 2. 5. 09:20

Anaconda Prompt에서 가상환경을 활성화한다.

conda activate test

tensorflow와 tensorflow-gpu를 설치한다.

pip install tensorflow==2.4.0
pip install tensorflow-gpu==2.4.0

ipykernel을 설치하고 jupyterlab에서 사용할 kernel을 설정한다.

pip install ipykernel

python -m ipykernel install --user --name test_venv --display-name "tensorflow2.4.0"
jupyter kernelspec list

※ kernel 삭제를 위해서는 아래 명령어를 사용한다.

jupyter kernelspec uninstall "tensorflow2.4.0"

jupyterlab에서 GPU 사용 가능 여부를 확인한다.

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

[참고]