본문 바로가기

dev_AI_framework

CUDA, CUPY, Python 버전 세팅 및 확인


📦 3. CuPy 설치


🧪 4. 테스트 코드 예시

 
import cupy as cp

x = cp.array([1, 2, 3])
y = cp.square(x)
print("GPU 연산 결과:", y)

📌 환경 전체 요약

Python 3.12.4
CUDA Toolkit 12.6
CuPy 13.4.1 (cupy-cuda12x)
가상환경 venv 사용 권장
설치 확인 nvcc --version, cupy.show_config() 등