RTX5050

Ubuntu 22.04 + RTX 5050에서 Docker로 TensorFlow sm_120 소스 빌드와 YOLO11 GPU 환경 구축하기 #2

by

24. YOLO11은 별도의 Docker 환경 사용 TensorFlow와 YOLO11을 같은 Docker에 모두 설치할 수도 있습니다. 하지만 여기에서는 분리합니다. 구조는 다음과 같습니다. TensorFlow는 TensorFlow 학습에 사용합니다. YOLO Docker는 PyTorch와 Ultralytics YOLO에 사용합니다. 이렇게 분리하면 문제가 발생했을 때 원인을 찾기 쉽습니다. 24. YOLO Dockerfile 만들기 다음 파일을 만듭니다. 다음 내용을 입력합니다. 파일을 저장합니다. 1) Ubuntu 기본 패키지 설치 … Read more

Ubuntu 22.04 + RTX 5050에서 Docker로 TensorFlow sm_120 소스 빌드와 YOLO11 GPU 환경 구축하기 #1

by

Ubuntu 22.04 노트북에서 NVIDIA GeForce RTX 5050 Laptop GPU를 이용해 TensorFlow와 YOLO11을 실행하려고 하면 기존 RTX 30, RTX 40 시리즈와는 조금 다른 문제가 발생합니다. RTX 5050은 NVIDIA Blackwell 계열이며 CUDA Compute Capability가 12.0, 즉 sm_120입니다. 일반적인 Python 가상환경에 TensorFlow를 설치하면 다음 단계까지는 정상적으로 진행될 수 있습니다. GPU가 검색되거나 CUDA 라이브러리가 로딩되는 것처럼 보여도 실제 Conv2D, … Read more