torch 버전 관리하기!

TOLL TERRY·2022년 2월 18일
0

error_or_download

목록 보기
4/16

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 다운로드 해야 한다.

profile
행복을 찾아서(크리스 가드너)

0개의 댓글