Microsoft Student (2023)728x90
파이토치(PyTorch)에서 연산 장치 설정하기 (GPU, CPU)
들어가며
- 파이토치(PyTorch)에서 연산 장치(GPU/CPU)를 설정하는 코드를 알아보자.
방법
- GPU 사용이 가능한 경우 @torch.device@를 @cuda@를 설정하고, 그렇지 않을 경우 @cpu@를 연산 장치로 설정한다.
# GPU 설정 (사용 가능한 경우)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("device >>" , device)
728x90
'Programming > Python' 카테고리의 다른 글
| [Python] 맵(Map)과 리스트(List)의 차이점 (0) | 2023.10.08 |
|---|---|
| [Python] try except 문을 사용할 때, 예외(Exception) 정보 출력하는 방법 (0) | 2023.09.08 |
| [Python] 삼항 연산자(Ternary Conditional Operator) (0) | 2023.08.23 |
| [Python] 딕셔너리에 리스트 형태의 값(Value)을 추가하는 방법 (0) | 2023.08.04 |
| [Python] 사이킷런(Scikit-Learn) 라이브러리 설치하기 (ModuleNotFoundError: No module named 'sklearn' 해결법) (0) | 2023.07.18 |
| [Python] 아나콘다(Anaconda) 프롬프트에서 ML/DL 실습 환경 세팅하기 (PyTorch) (0) | 2023.07.12 |
| [Python] 파이썬 코드를 실행 파일로 만들기 (PyInstaller) (0) | 2023.07.03 |
| [Python] 아나콘다 프롬프트를 이용하여 가상 환경 만들기 (0) | 2023.06.12 |