본문 바로가기

dev_AI_framework

CUDA C++ 코드의 Python 호출 - 1. CUDA, pytorch 버전 맞추기

1. CUDA C++ 파일 작성

2. CMake 파일 작성 : CMake 를 사용해 CUDA 와 Pybind11 을 함께 빌드,

3. 빌드 스크립트 작성 : 빌드 관리

 

빌드 스크립트 코드 작성 중 torch 모듈 import 문제 발생, 

 

CUDA 와 torch 버전 문제 해결 필요

CUDA의 버전을 torch 에서 제공하는 버전으로 맞춤

와 근데 이거 말고도 python버전 문제도 있었음, 3.9 버전 이상이면 다 되는 줄 알았는디... 

3.11 버전에선 안먹었음, python 버전도 3.12 로 변경 후 해당 문제 해결...