반응형
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
반응형
'Python > PyTorch 공부' 카테고리의 다른 글
| [PyTorch] torch.nn.init.xavier_uniform 가중치 초기화 (2) | 2021.11.02 |
|---|---|
| [PyTorch] Boolean value of Tensor with more than one value is ambiguous (0) | 2021.10.31 |
| [PyTorch] nn.Sequential 을 상속받아 Class 정의하기 (0) | 2021.08.04 |
| 이미지 분류 신경망의 결과를 t-SNE 시각화하기 (0) | 2021.07.11 |
| [PyTorch] ShuffleSplit와 subset 함수를 사용하여 dataset 분할하기 (0) | 2021.07.10 |