Trouble/Python, Pytorch 13

torch.cdist 사용 시 유의점

torch.cdist 는 document에 따르면 다음과 같은 기능을 한다. 요약하자면, tensor와 tensor 간의 distance matrix를 반환해주는 함수라고 할 수 있다. 대표적으로 3D points 간 거리 계산에 유용하게 쓰인다. x,y,z로 표현 되는 3D point 집합 X와 Y가 있고 각각 N개, M개의 3D point로 구성되어 있다고 했을 때, X와 Y 집합 간 서로 최단 거리로 이웃한 점들을 선별할 때나 특정 거리 이내의 이웃한 점들은 선별할 때 자주 사용하게 된다. 하지만, 가끔 제대로 기능하지 않는 듯한 모습을 보일 때가 있어 이를 정리하고자 한다. "가끔 제대로 기능하지 않는 듯한 모습"은 크기가 25 이상인 경우에 발생했다. 위 document에 따르면 P나 R이 25..

torch_geometric/pyg 사용 시 propagate 혹은 forward 오류 해결법

Graph neural network 구현을 용이하도록 개발된 torch_geometric 혹은 요즘 새로운 이름 pyg를 사용할 때, 모든 오류를 고쳐서 가까스로 import 했더니 propagate나 forward 에서 터져버리는 오류가 날 때가 있다. 온갖 구글링 결과, 같은 상황으로 고통받던 한 사람이 단순히 open3d와 torch의 import 순서를 바꿨더니 해결했다는 말이 있었다. 다시 말하면, 왜 그런지 모르겠는데 open3d가 torch 이전에 import 되어있을 때 났다 import open3d import torch # this order causes torch_geometric error 위와 같은 순서로 하면 오류가 나고 import torch import open3d # th..

RTX 3090 GPU pytorch error : cuda 사용 안되는 문제

RTX 3090 GPU를 장착한 컴퓨터에서 conda나 pip로 pytorch를 그냥 설치할 경우, 아래와 같이 cuda 사용이 안된다는 오류가 뜰 때가 있다. RTX A6000 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70. If you want to use the RTX A6000 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/ 이유는 적혀있는대..