에러 원인 : GPU에 할당되어 있는 tensor를 Numpy 배열로 변환할 때 생기는 에러
- gpu에 할당된 tensor를 .numpy()로 변경을 하려고 할 때 생김
- .cpu( )를 사용하여 gpu에 할당된 tensor를 cpu tensor로 변환
# if X is CPU tensor
X.cpu().numpy()
반응형
'Language > python' 카테고리의 다른 글
| [Python] 파일 확장자명 일괄변경 (0) | 2023.04.27 |
|---|---|
| [Python] 리스트 안의 원소들을 한번에 int type 으로 변환 (0) | 2023.02.14 |
| [Python] Image 읽는 방법 (0) | 2023.02.10 |
| [numpy] numpy 소수점 반올림하기 / array 전체 출력 / 출력 형식 변경 (0) | 2023.01.25 |
| [matplotlib] matplotlib 그래프 그리기 (0) | 2023.01.17 |
