search

haven-jeon / KoGPT2-chatbot

Simple Chit-Chat based on KoGPT2
MIT License
180 stars 63 forks source link

segmentation fault 오류 #22

Open minkyung73 opened 1 year ago

minkyung73 commented 1 year ago

코드를 실행하다가 오류가 나서 여쭤봅니다.

m1 로컬 환경에서 시도를 하였고, 이를 위해 파이토치 공식 홈페이지에 나와있는 코드(conda install pytorch torchvision torchaudio -c pytorch)로 설치를 진행하였습니다.

터미널에 CUDA_VISIBLE_DEVICES=0 python train_torch.py --gpus 1 --train --max_epochs 2를 입력하면 아래와 같은 오류가 나는데 해결 방법이 있을까요? zsh: segmentation fault CUDA_VISIBLE_DEVICES=0 python trainer_s.py --gpus 1 --train --max_epochs 2

minkyung73 commented 1 year ago

python train_torch.py --train으로 해도 동일한 오류가 발생합니다.

버전은 transformers==4.11.3 torch==1.8.1 pytorch-lightning==1.2.10 입니다.

siyeol97 commented 1 year ago

m1은 cuda 를 쓸수 없는걸로 아는데용