Ubuntu에서 nvidia-smi 실행시 Failed to initialize NVML: Driver/library version mismatch 오류 해결하기

Pytorch를 사용할 때, torch.cuda.is_available()가 False 가 나오면, GPU 연결에 문제가 생겼을 가능성이 높다. import torch import torch.nn.functional as F if torch.cuda.is_available(): device = 'cuda' else: device = 'cpu' 터미널에 nvidia-smi 를 쳐서 확인해보자. $ nvidia-smi Failed to initialize NVML: Driver/library version mismatch 확인해보니, Failed to initialize NVML: Driver/library version mismatch 오류가 발생했다. 재부팅하는 방법도 있지만, 재부팅을 하지 않고서 해결..

2023.07.30