냥냥파워

RuntimeError: cuda runtime error(10): invalid device ordinal 본문

POSTECH AI연구원 인턴

RuntimeError: cuda runtime error(10): invalid device ordinal

hjhjhj0028 2022. 6. 17. 15:47

https://intrepidgeeks.com/tutorial/pythorc-reading-parameter-error-runtimeerror-cuda-runtimeerror-10-invalid-device-serial-number

 

Pytorch 읽기 매개 변수 오류 RuntimeError: cuda runtime error(10): invalid device ordinal

MXNet에서 시각적 네트워크 구조 내보내기 시각 형상 내보내기 오류: ndarray, x = nd.random.uniform(shape=(1,3,224,224),ctx=mx.gpu()) 이것은 MXNet이 동적 그림의 forward 계산에 따라 정적 그림을 생성하는 절차 문

intrepidgeeks.com

https://stackoverflow.com/questions/22175825/cuda-invalid-device-ordinal

 

CUDA: invalid device ordinal

I have the following problem. I want to allow my users to choose which GPU to run on. So I was testing on my machine which has only one GPU (device 0) what would happen if they choose a device which

stackoverflow.com

사용 가능한 gpu가 여러대일 때

gpu id 할당을 잘못하면

생기는 에러다

 

나는 현재 NVIDIA Quadro K620

NVIDIA GeForce GTX 1080 Ti

2대가 잡히기 때문에

사용할 gpu id를 지정해서 해결했다

 

 

윈도우 유저라면 

ctrl + shift + esc를 눌러서

작업 관리자>성능 탭을 보면

사용 가능한 GPU를 확인 할 수 있다

 

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

 

BIG

'POSTECH AI연구원 인턴' 카테고리의 다른 글

ViT for Image Classification (Visual Transformer)  (0) 2022.06.21
Image Classification with ResNet50 in Pytorch  (0) 2022.06.20
YoloV5 Detector로 OCR 데이터셋 한글 검출하기  (0) 2022.06.16
RuntimeError: CUDA error: device-side assert triggered  (0) 2022.06.16
YOLACT : Real Time instance segmentation  (0) 2022.06.07
'POSTECH AI연구원 인턴' Related Articles more