코드민수
[Python] Tensorflow & Pytorch GPU 사용 확인 본문
BIG
tensorflow에서 GPU 사용 가능 여부 확인
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
Pytorch에서 GPU 사용 가능 여부 확인
import torch
print(torch.cuda.is_available())
CUDA, cudnn, tensorflow, pytorch가 버전에 맞게 잘 설치되어 있어야 작동합니다.
LIST
'[Python] > 코드' 카테고리의 다른 글
| [Python] plt.savefig() 시 생기는 여백 제거 방법 (0) | 2023.05.10 |
|---|---|
| [Python] 경로 내 이미지 전체의 평균, 표준편차 구하기 (0) | 2023.05.06 |
| [Python] 레이블 없는 이미지 찾기 (0) | 2023.04.03 |
| [Python] JSON 파일에서 오브젝트 클래스 분석 (히스토그램) (0) | 2023.03.31 |
| [Python] 중복 이미지 제거 (0) | 2023.03.30 |
'[Python]/코드' Related Articles
more
