티스토리 뷰
Tensor형태의 데이터는 .is_cuda를 붙여보면 cuda위에 있을경우 True, cpu에 있을 경우 False를 반환한다.
따라서 확인하기 쉽지만 class instance인 model은 단순하게 model.is_cuda를 할 경우 attribute error가 난다.
모델이 cuda에 있는지 확인하기 위해서는 아래 코드를 입력하면 된다.
next(model.parameters()).is_cuda
(여기서 model은 본인의 모델을 저장한 변수 이름으로 바꿔야한다.)
'Computer > 코딩 개꿀팁' 카테고리의 다른 글
| 딥러닝 모델을 돌리는데 램 누수 현상이 발생한다? (0) | 2023.04.12 |
|---|---|
| [꿀팁]파이썬을 이용해서 파일명을 일괄로 변경해보자 (0) | 2023.03.05 |
| Pytorch tensor를 더 많이/정확하게 표기하고 싶을 때 (0) | 2023.02.23 |
| f.write에서 변수를 사용하려면 repr()함수 (0) | 2023.02.03 |
| [꿀팁] str을 dictionary로 간단히 바꾸는 eval함수 (0) | 2023.02.02 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
TAG
- 오일석기계학습
- 백준
- eigenvalue
- manim
- 3b1b
- nanobody
- 인공지능
- 최대우도추정
- variational autoencoder
- 3B1B따라잡기
- Manimlibrary
- 파이썬
- 기계학습
- 베이즈정리
- 나노바디
- MatrixAlgebra
- eigenvector
- kl divergence
- elementry matrix
- MLE
- Matrix algebra
- MorganCircularfingerprint
- 제한볼츠만머신
- marginal likelihood
- ai신약개발
- 선형대수
- manim library
- manimtutorial
- kld
- 이왜안
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | ||||||
| 2 | 3 | 4 | 5 | 6 | 7 | 8 |
| 9 | 10 | 11 | 12 | 13 | 14 | 15 |
| 16 | 17 | 18 | 19 | 20 | 21 | 22 |
| 23 | 24 | 25 | 26 | 27 | 28 | 29 |
| 30 | 31 |
글 보관함