베이스라인 학습 시 오류 나시는 분들께 CUDA error: CUBLAS_STATUS_INVALID_VAL

도배 하자 질의 응답 처리 : 한솔데코 시즌2 생성 AI 경진대회

터틀

2024.01.29 12:22 1,995 Views

pip install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

pip install transformers==4.19.2

가상환경 새로 만들어서 위 버전으로 설치해보시길 pip install 해보시길 바랍니다.

Linux & Windows - Conda:

# CUDA 10.2
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

# CPU Only
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cpuonly -c pytorch

-> cuda 11.X버전 사용하시는 분들 11.3 버전으로 설치해주세요.

파이토치 버전 안 맞으시면 아래 사이트에서 1.11.X 설치해보세요.

https://pytorch.org/get-started/previous-versions