반응형 .cpu()1 [에러] TypeError : can't convert CUDA tensor to numpy 에러 원인 : GPU에 할당되어 있는 tensor를 Numpy 배열로 변환할 때 생기는 에러 gpu에 할당된 tensor를 .numpy()로 변경을 하려고 할 때 생김 .cpu( )를 사용하여 gpu에 할당된 tensor를 cpu tensor로 변환 # if X is CPU tensor X.cpu().numpy() 2023. 2. 9. 이전 1 다음 반응형