728x90
반응형
먼저 https://kc9302.tistory.com/114 이걸 참고해서 1~4번까지 하고 나서 다음 내용을 확인해볼 수 있다.
구분 | 버전 |
GPU | GeForce GTX 1050 Ti |
Anaconda | Anaconda3-2023.09-0-Windows-x86_64 |
Python | 3.8 |
CUDA | cuda_11.2.0_460.89_win10 |
tensorflow-gpu | 2.5.0 |
torchvision | 0.9.0 |
torchaudio | 0.8.0 |
cudatoolkit | 11.1 |
이전에 설정한 환경과 동일하다. 하지만 처음 동작시 GPU를 인식하지 못했다.
원인은 tensorflow 버전이 맞지 않았다.
https://spltech.co.uk/installing-tensorflow-2-9-with-gpu-support/ (이곳에서 버전에 맞는 환경 설정을 확인할 수 있다.)
CUDA 11.2를 설치했으니 Tensorflow 2.5.0을 설치한다. 여기서 Tensorflow를 설치하는 게 아닌 Tensorflow-GPU로 설치해준다.
conda install tensorflow-gpu==2.5.0
상황에 따라 pip로 설치해도 문제없다.
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
이후 이 코드를 실행했을 때 아래와 같이 GPU가 잡힌다면 올바르게 환경을 설정한 것이다.
반응형
'데이터분석 > Python' 카테고리의 다른 글
넘파이 배열 생략없이 (0) | 2024.05.03 |
---|---|
경로 탐색 라이브러리 os 보다 편리한 glob라이브러리 (0) | 2024.03.13 |
Window Pycharm(파이참) GPU 연결 with pytorch (2) | 2023.11.26 |
[Python] code formater, 코드 정리 라이브러리 Black (0) | 2023.07.07 |
[Python] import time, 시간 측정 (0) | 2023.06.16 |
댓글