https://pytorch.kr/get-started/previous-versions/
토치사이트 버전 torchvision, torchaudio
자신의 현재 버전 확인하기 torch가 만약 깔려져 있다면 확인용
!python -c 'from torch.utils.collect_env import main; main()'
버전 확인 주피터에서 명령에서 확인 !
locate cuda-toolkit
# 명령어는 터미널에서 그냥 치면 됩니다.
현재 자신의 가상 환경의 쿠다 버전을 확인해주세요 (저는 10.1)
torch를 cuda-toolkit / 10.1에 맞게 설치하신 후
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
import torch
USE_CUDA = torch.cuda.is_available()
print(USE_CUDA)
device = torch.device('cuda:1' if USE_CUDA else 'cpu')
print('학습을 진행하는 기기:',device)
torch.cuda.get_device_name(0)
torch가 잘 되었는지 확인합니다.
다음에 torch-geometric를 설치해야 합니다.
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.7.1+cu101.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.7.1+cu101.html
pip install torch-geometric
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.7.1+cu101.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.7.1+cu101.html
저는 여기서 참조했습니다.
https://github.com/pytorch/vision#installation
torch // torchvision
torch // torchtext
이런데 갑자기 에러가 떳다 ..
OSError: /home/user/anaconda3/envs/hs/lib/python3.7/site-packages/torchsparse/_version_cuda.so: undefined symbol: _ZN3c106detail12infer_schema20make_function_schemaENS_8ArrayRefINS1_11ArgumentDefEEES4
[출처](https://github.com/pyg-team/pytorch_geometric/issues/3575)
해결방법:
torch_geometric
버전 다운로드 시에 nvidia-smi로 확인 후
각자의 가상환경에 들어가서 torch 깔고, torch-geometric 다운로드 해야 한다.