ufris
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1! (when checking argument for argument weight in method wrapper_CUDA__cudnn_convolution) 에러 해결 본문
딥러닝
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1! (when checking argument for argument weight in method wrapper_CUDA__cudnn_convolution) 에러 해결
ufris 2025. 2. 20. 16:01gpu 0 만 사용해서 잘 돌아가다가 다른 학습을 위해 gpu 1번을 사용하는 순간 해당 에러가 발생
학습이 잘 되던 gpu 0 도 오류가 발생해서 학습이 진행 안됐는데
$ torch.cuda.init() # CUDA 시스템 초기화
코드가 있으면 위와 같은 오류가 발생하기 때문에 해당 코드는 삭제 후 돌리니까 오류가 해결 됐습니다
'딥러닝' 카테고리의 다른 글
| 학습 도중 killed 문제 (0) | 2025.02.18 |
|---|---|
| UMA node read from SysFS had negative value (-1) (0) | 2022.01.07 |
| [TF2] Output Class 개수가 다를 때 pretrained model transfer learning (0) | 2021.07.30 |
| Loss와 Accuracy는 항상 반비례 관계인가 (0) | 2021.07.15 |
| [Cuda] GTX 3090 사용 시 tensorflow graph 생성이 느린 문제 (0) | 2021.01.20 |
'딥러닝' Related Articles
more
