[Python_Error] NVIDIA GeForce RTX 3070 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation. 해결

회사 노트북이 RTX 3070인데

원격으로 접속해서 사용하려고하니 에러가 떴다!

 

에러 내용은 

NVIDIA GeForce RTX 3070 Laptop GPU 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_70. If you want to use the NVIDIA GeForce RTX 3070 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

이고 

이것은 pytorch 버전이 맞지 않아서 발생하는 것인 걸 알아냈다!

 

무슨 뜻이냐면

노트북에 깔려있는 cuda 버전은 11.6인데

(cuda 버전은 nvidia-smi 코맨드로 확인)

pytorch 버전이 cuda 11 버전에 호환되는 애가 아니라서 그런 것!!

 

아마 디폴트가 10버전에 맞게 설치된다고 한다.

 

그래서 

링크에 들어가서

이렇게 설정한 후 밑에 나온 코멘드를 터미널에 입력하여 설치하면 된다.

 

그런데 여기서 주의해야할 점!

사용하려는 프로젝트의 가상환경을 활성화 한 후 설치해야함!

 

어쩌고저쩌고 

설치가 완료 된 후에

다시 실행해보면~!

잘 될 것이다!!

 

만약 이랬는데 잘 안된다! 싶으면

잘 설치가 된 건지 확인해보자

 

나의 경우

기존에 torch 1.8.0버전을 사용했었는데

python

import torch

torch.__version__

으로 버전을 확인해 본 결과 1.8.0 이었고

이 때 오류가 난 것!

 

위처럼 새로 설치를 완료한 후 확인해보면

이런식으로 뒤에 cu113 이 붙어서 나온다!

이러면 설치가 잘 된 것.

---

그럼 끝!