pytorch 일반적인 오류

1 RuntimeError: CUDA 오류: 장치에서 실행할 수 있는 커널 이미지가 없습니다.

오류 메시지: GPU와 함께 Pytorch를 설치할 때
(1) 먼저
CUDA 기능이 있는 NVIDIA GeForce RTX 3090 sm_86이 현재 PyTorch 설치와 호환되지 않는다고 경고합니다. 현재
PyTorch 설치는 CUDA 기능 sm_37 sm_50 sm_60 sm_70 sm_75을 지원합니다
. PyTorch가 포함된 NVIDIA GeForce RTX 3090 GPU, https://pytorch.org/get-started/locally/에서 지침을 확인하세요.
(2) 그런 다음 딥 러닝 코드를 실행할 때 오류가 보고됩니다:
RuntimeError: CUDA error: no kernel image 장치에서 실행할 수 있습니다.

이유: pytorch1.6 버전은 GPU 하드웨어 RTX 3090을 지원하지 않기 때문에 더 높은 버전의 pytorch를 다운로드해야 합니다.

해결 방법: pytorch 버전 1.8 이상을 다운로드하고 컴퓨팅 성능이 이를 지원해야 합니다.

// 查看包支持的算力
 import torch
 torch.cuda.is_available()
// 上面返回true
 torch.cuda.get_arch_list()

Torch는 GPU 하드웨어에 해당합니다.

행복한 고민: 그래픽 카드의 연산 능력이 너무 높고 pytorch 버전이 너무 낮아서 지원하지 않습니다_cuda 버전이 너무 높아 pytorch가 지원하지 않습니다_Yu의 블로그입니다 - CSDN 블로그