728x90
study-grow.tistory.com/entry/multi-gpu-%EC%8B%A4%ED%8C%A8-tensorflow-mirrored-strategy
multi gpu 실패, tensorflow mirrored strategy
멀티 GPU가 잡히지 않을 떄 mirrored_strategy = tf.distribute.MirroredStrategy() tf.keras로 multi gpu를 쓰려면 위에 구문을 쳐주고 mirrored_strategy scope내에 model 생성 및 compile을 해줘야 한다. with..
study-grow.tistory.com
위 방법을 쓰면 될 줄 알았다만... gpu가 동작하지 않는다... 뭐가 문제인 걸까?
NotImplementedError: _reduce method must be implemented in descendants.
이렇게 하면 작동하긴 하는데...
nvidia-smi를 쳐보면 여전히 gpu가 작동 안 한다.
tensorflow 버전과 cuda 버전이 안 맞는 게 아닐까? tf 버전을 낮춰보자.
'Data-science > deep learning' 카테고리의 다른 글
| MBTI 성격유형별 관상 stylegan2 morphing image (2) | 2020.12.22 |
|---|---|
| 동물(개, 강아지) 얼굴만 추출, pretrained detection model 이용하기 (0) | 2020.12.20 |
| multi gpu 실패, tensorflow mirrored strategy (0) | 2020.12.12 |
| dacon - Rainnet기반 강수량 예측시 CSI, MAE 계산 (0) | 2020.11.14 |
| Spatiotemporal CNN for Video Object Segmentation (0) | 2020.11.04 |
