728x90
버전 확인
import torch
print(torch.__version__)
GPU 연결 확인
device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(device)
728x90
'Python > numpy & Pytorch' 카테고리의 다른 글
| csv 파일 json으로 바꾸기 (0) | 2024.03.29 |
|---|---|
| [Numpy] 넘파이 난수 생성하기 (0) | 2023.05.04 |
| [Data] torchvision datasets으로 받은 데이터 나누기 (0) | 2023.03.27 |
| [Pytorch] Tensor shuffle, 텐서 랜덤 섞기 (0) | 2023.03.16 |
| [Pytorch] tensor 합치기는 방법 cat(), stack() (0) | 2023.03.14 |
728x90
버전 확인
import torch
print(torch.__version__)GPU 연결 확인
device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(device)
728x90
'Python > numpy & Pytorch' 카테고리의 다른 글
| csv 파일 json으로 바꾸기 (0) | 2024.03.29 |
|---|---|
| [Numpy] 넘파이 난수 생성하기 (0) | 2023.05.04 |
| [Data] torchvision datasets으로 받은 데이터 나누기 (0) | 2023.03.27 |
| [Pytorch] Tensor shuffle, 텐서 랜덤 섞기 (0) | 2023.03.16 |
| [Pytorch] tensor 합치기는 방법 cat(), stack() (0) | 2023.03.14 |
