Python/PyTorch 공부
[에러 해결] CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)
AI 꿈나무
2021. 9. 14. 20:53
반응형
pytorch lightning으로 분산학습시에 오류가 발생했다.
CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)` 오류 발생시 해결방법은 pytorch 버전과 cuda 버전을 맞춰줘야 한다 !@
아래 코드를 입력하면 최신 버전 torch로 업데이트 해준다. 하하하
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
출처
CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)`
That’s expected, since the pip wheels and conda binaries ship with their own CUDA runtime. Your local CUDA toolkit (shown via e.g. nvidia-smi) would be used if you are building a custom CUDA extension or PyTorch from source. Thanks for the env informatio
discuss.pytorch.org
반응형