1. 문제 설명

맞춤 C++ 작업을 로드할 수 없습니다. PyTorch와 torchvision 버전이 호환되지 않는 경우에 발생할 수 있습니다.
이 문제가 처음 발생했을 때 torch와 torchvision의 버전이 일치하지 않는다고 합니다. 하지만 저는 이 환경을 오랫동안 사용해 왔으며 아무런 문제가 없었습니다( conda와 함께 설치됨 ). 하지만 문제가 발생했으므로 해결 방법을 찾아야 합니다!

2. 문제 분석

2.1 토치 및 토치비전 버전 보기

다음 코드를 사용하여 Pycharm에서 볼 수 있습니다.

import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
print(torch.version.cuda)

산출:

1.12.1
0.13.1
11.3

토치와 토치비전의 몇 가지 버전을 읽어본 결과 문제가 없다는 것을 알았습니다.

2.2 토치, 토치비전, 토치오디오 재설치

인터넷에 대한 몇 가지 의견과 결합하여(인수 1: torch, torchvision, torchaudiion 제거하고 해당 버전 다시 설치; 인자 2: pip를 사용하여 설치하는 것이 좋습니다) 따라서 먼저 모든 torch, torchvision, torchaudion을 제거한 다음 pip로 다시 설치하십시오.

2.2.1 토치, 토치비전, 토치오디온 제거

pip uninstall torch
pip uninstall torchvision
pip uninstall torchaudio

conda uninstall torch
conda uninstall torchvision
conda uninstall torchaudio

이제 다운로드가 깨끗합니다

2.2.2 핍 설치

pytorch 공식 웹사이트에서 자신의 cuda에 해당하는 torch, torchvision, torchaudio 버전을 찾아 pip로 설치하도록 선택합니다
.

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

설치에 성공하면 다음 버전을 볼 수 있습니다.

import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
print(torch.version.cuda)

산출:

1.12.1+cu113
0.13.1+cu113
11.3

처음부터 조금 다른 것 같지만 그 차이가 그다지 크지 않다는 느낌이 듭니다.
어쨌든 마지막 문제는 해결되었습니다. 오류는 Couldn't load custom C++ ops입니다.

그러나 PIL 모듈을 찾을 수 없다는 오류 메시지가 표시되지만 이것은 작은 문제이므로 제거하고 다시 설치하면 됩니다.

pip uninstall pillow
pip install pillow

코드가 잘 작동하고 더 이상 문제가 없습니다!

3. 요약

따라서 pip 설치에는 문제가 없지만 conda 설치에는 문제가 있으므로 pip 설치를 권장합니다.

맞춤 C++ 작업을 로드할 수 없습니다.