728x90 반응형 PyG4 [conda] [PyG] torch geometric 설치 오류 (오래된 cuda version) Conda에서 pyg 설치하기 (CUDA 버전 맞추기)연구실 서버를 고쳐야 하는데 실험을 중단할 순 없으니 학과 서버를 대여하기로 했다.그래서 개발환경을 다시 설정해야하는데...연구실 GPU의 CUDA version이 달라서 완전히 똑같은 세팅을 할 순 없었다. Update: 2024.11.06 (CUDA 10.1)아니 이번에는 cuda 버전이 10.1이다더 낮은 버전이다....... 미치겠다하루종일 또 시간 걸려서 겨우 설치했다cuda 10.1이 되는 pytorch를 1.8.1로 설치하고 나머지는 아래 참고[3] 덕분에 해결했다(pyg_env) $ python -c "import torch; print(torch.__version__)"1.8.1+cu101(pyg_env) $ python -c "imp.. 2024. 7. 22. [PyG] GIN 예제 코드 Introduction GIN은 graph-level task에 적합한 GNN 모델이다. protein dataset을 이용하여 graph classification을 수행해보자. Setup import gc import os import random import numpy as np import torch import torch.nn as nn import torch.nn.functional as F from torch_geometric.nn import GCNConv, SAGEConv, GATConv, GINConv, global_add_pool, global_mean_pool from torch_geometric.datasets import TUDataset from torch_geometric.l.. 2023. 7. 12. [PyG] GCN, GraphSAGE, GAT 예제 코드 Import import gc import os import random import numpy as np import torch import torch.nn as nn import torch.nn.functional as F from torch_geometric.nn import GCNConv, SAGEConv, GATConv from torch_geometric.datasets import Planetoid Load Dataset (Cora dataset) os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="1" device = torch.device('cuda:1' if torch.cuda.is_avail.. 2023. 7. 12. [CS224w] Colab 2 - PyG, OGB, GNN Device 런타임 > 런타임 유형 변경 > 하드웨어 가속기 > GPU로 설정하고 저장 Setup import torch import os print("PyTorch has version {}".format(torch.__version__)) PyTorch has version 1.13.1+cu116 # Install torch geometric if 'IS_GRADESCOPE_ENV' not in os.environ: !pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.13.1+cu116.html !pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.13.. 2023. 3. 21. 이전 1 다음 728x90 반응형
