torch 정상 동작 검사, 3080 그래픽카드 torch gpu 문제 해결

"

GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.

 

 

 

 

3080 그래픽카드에서 pytorch 가 동작하지 않는 문제가 간혹 발생합니다.

해결 방법은 간단합니다.

cuda toolkit version 을 11.0 이상으로 올려줍니다.

 

cuda 설치 방법 안내는 아래 링크를 확인하세요.

https://dokim.tistory.com/217

cuda 설치 + cudnn 다운 + pytorch 설치 + torch 설치

 

 

설치 후 정상 동작 확인을 위해서는 아래 명령어를 통해 알 수 있습니다.

>>> import torch

>>> torch.cuda.get_device_name(0)

.

>>> import torch >>> torch.cuda.is_available()

2. U

일반적으로 아래와 같이 사용 가능한지에 대한 함수를 이용하는데, 이때에는 에러 메세지가 나타나지 않습니다.

위와같이 get_device_name(0) 을 실행하였을 때, 현재 cuda 를 이용할 수 있는지에 대해서 확인이 가능합니다.

 

>>> torch.cuda.is_available()