python, pyTorch

torch-geometric install error(2)

zooyeonii 2021. 8. 6. 17:49

나의 PC:
GEFORCE RTX 2060 (notebook)
Windows10
Python 3.7
PyTorch 1.8.1
CUDA 11.1 
cuDNN 8.0.4 


torch-geometric doc를 확인하니 몇가지 조건이 있었다.
(1) Windows 에서는 CUDA 10.2 이상을 요한다. 
(2) PyTorch version 1.8.0 이상 
(3) OS/Pytorch version/CUDA version 에 맞게 선택해야 한다.

CUDA 설치CUDA 재설치는 전 게시글 참고. 

'python, pyTorch' 카테고리의 다른 글

GNN dataset (dynamic graph)  (0) 2021.08.11
tensorflow-gpu 에러 (해결)  (0) 2021.08.07
torch-geometric install error(1) CUDA 재설치  (0) 2021.08.06
GPU_pytorch 사용하기  (0) 2021.08.06
python 코딩 : list 거꾸로 출력하기  (0) 2021.07.02