[Kaggle] Kaggle은 PyG를 설치합니다 | 설치한 라이브러리 파일을 추출합니다.

1. 문제 설명

pip를 사용하여 직접 다운로드 및 설치할 수 있지만 Kaggle의 BUG로 인해 일부 라이브러리는 온라인 설치 속도가 느려집니다.

PyG를 예로 들면 온라인 설치는 30-60分钟좌우가 필요한 경우가 많은데 필요한 휠 파일을 Kaggle에 업로드하고 로컬 설치를 이용하면 설치가 1分钟완료된다.

---

2. 해결 단계

2.1 기본 환경 제거

우선 pyg는 torch와 cuda 버전에 대한 요구 사항이 있지만 kaggle의 기본 환경은 pyg를 지원하지 않으므로 먼저 kaggle과 함께 제공되는 torch를 제거합니다.

pip uninstall torch torchvision torchaudio --y

2.2 적절한 버전 설치

토치를 설치합니다. 권장 버전은 1.12.1-cu113입니다.

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

2.3 PyG 설치

pyg 설치

처음으로 kaggle의 GPU 가속을 켜고 다음을 실행합니다.

pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv torch_geometric -f https://data.pyg.org/whl/torch-1.12.0+cu113.html

오랜 시간을 기다린 후 설치가 성공적으로 완료되었다는 메시지가 표시됩니다.

그것을 테스트, 출력은 Data(edge_index=[2, 4], x=[3, 1])설치가 성공적이라는 것입니다

import torch
from torch_geometric.data import Data

edge_index = torch.tensor([[0, 1, 1, 2],
                           [1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)

data = Data(x=x, edge_index=edge_index)
print(data)

2.4 설치 캐시 휠 파일 추출

저장 디렉토리 생성

import os
os.mkdir('files')

파일을 출력 폴더에 복사한 다음 출력에서 ​​다운로드할 수 있습니다.

!cp -r  ~/.cache/pip/wheels/*/*/*/*/*.whl /kaggle/working

다운로드가 완료되면 이 whl 파일을 폴더에 넣은 후 다음 두 whl 파일을 다운로드합니다(다운로드하려면 클릭).

• pyg_lib
• torch_sparse

위에서 다운로드한 whl 파일 6개를 모두 압축 패키지로 압축하고, kaggle에서 데이터셋을 생성한 후 압축 패키지를 업로드합니다.

pyg를 사용해야 하는 노트북에 압축된 패키지를 추가하기만 하면 됩니다.

2.5 압축 패키지에 whl 파일 설치

!pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv torch_geometric --no-index --find-links=file:///kaggle/input/pyg-packages-torch1121-cu113/pyg-packages

내 데이터 세트를 직접 사용할 수도 있습니다. 이름은pyg-packages-torch1.12.1+cu113