[PyTorch]GPU와 CUDA 호환 확인하기

Python/PyTorch 공부

AI 꿈나무 2021. 11. 25. 00:06

torch.cuda.current_device()로 확인할 수 있다.

위 사진을 보면 A6000는 CUDA 10.2 버전과 호환이 안된다.

따라서 CUDA 11.3 버전을 다시 설치해줘야 한다. 참고로 파이토치도 함께 재설치를 해줘야된다.

CUDA 버전을 업그레이드하니 작동된다.

[PyTorch] RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR (1)	2021.12.03
[PyTorch] Error: one of the variables needed for gradient computation has been modified by an inplace operation (1)	2021.11.25
[Pytorch] nn.Unsample에서 align_corners=False? True? (0)	2021.11.18
[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

딥러닝 공부방