tensorflow 설치시 주의해야 한다.
# Anything above 2.10 is not supported on the GPU on Windows Native
python -m pip install "tensorflow<2.11"
# Verify the installation:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
2024년 4월 현재 윈도우에서는
pip install "tensorflow<2.11" 로 설치해야 한다.
import os, sys
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
한다음 import tensorflow as tf 로 불러와야 한다는데.....
그런데 막상 실행해보니
import os, sys
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
로 설정하면 처리속도가 빨라진다.
댓글 없음:
댓글 쓰기