pytorch2 [Docker] Dockerfile 작성을 통한 Docker Pytorch 환경 구성 * 목표- 특정 pytorch 버전이 적용된 Docker 환경을 구성합니다.- 파이토치 뿐 아니라 다양한 필요한 패키지들을 사용하기 위해 Dockerfile을 작성합니다.- 이번 포스트에서는 cuda 12.6, cudnn 9와 pytorch 2.9.1 로 구성된 도커 이미지를 직접 Dockerfile로 구성하겠습니다. * 용어 정리 Dockerfile: 도커 이미지를 만드는 “레시피”Image(이미지): 실행 가능한 파일시스템 스냅샷(템플릿)Build(빌드): Dockerfile에서 Image를 생성Container(컨테이너): Image를 실행한 인스턴스Run(실행): Image를 통해 Container를 생성하고 프로세스를 실행 1. 도커 설치1.1 오래된 버전 도커/충돌 패키지 제거 sudo ap.. 2026. 1. 24. [Pytorch] "RuntimeError: CUDA error: an illegal memory access was encountered" 에러 발생 시 * 에러 파이토치에서 텐서를 gpu로 옮겨주는 과정인 .to(device)를 실행하면 아래와 같은 에러 발생 에러가 어디서 발생했는지 확인하기 위해, 다음 환경변수를 설정 CUDA_LAUNCH_BLOCKING=1 * 해결 방법 터미널에 다음 명령어 입력 export TORCH_CUDNN_V8_API_DISABLED=1 * Reference https://github.com/pytorch/pytorch/issues/99372 PyTorch 2.0.0 encountered CUDA error: an illegal memory access was encountered · Issue #99372 · pytorch/pytorch 🐛 Describe the bug Running PyTorch 2.0.0 encoun.. 2023. 12. 12. 이전 1 다음
