코딩 메모장

  • 홈
  • 태그
  • 방명록

CUDA 2

[Docker] Container 내부에서 GPU 사용하는 방법

안녕하세요, 오늘은 Docker Container 내부에서 GPU를 간단하게 사용할 수 있는 방법에 대해 소개하려고 합니다. 총 2가지를 진행하면 됩니다. 간단하게 nvidia-docker 설치와 같은 방법을 사용하지 않고 진행합니다. Dockerfile 환경변수 추가 설정 Docker Container 실행시 매개변수 추가 Dockerfile 환경변수 추가 설정 # Dockerfile FROM python:3.8 ADD . /workdir RUN apt-get update && apt-get install vim -y RUN pip install torch && pip install transformers && pip install gpustat ##### 다음 줄 추가 ##### ENV CUDA_VI..

Program/Docker 2024.04.10

[PyTorch] RuntimeError: The NVIDIA driver on your system is too old. 오류 해결법

오늘은 GPU 버전 오류에 대해 포스팅을 하려고 합니다. 최근에 GPU를 이용해서 인공지능 모델을 학습시키려고 하는데 다음과 같은 오류가 발생했습니다. RuntimeError: The NVIDIA driver on your system is too old (found version 11040). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with you..

AI & DL/Pytorch 2023.12.31
이전
1
다음
더보기
250x250
프로필사진

코딩 메모장

  • 분류 전체보기 (149)
    • Paper (25)
      • Metric (6)
      • Model (3)
    • AI & DL (23)
      • Pytorch (6)
      • Hugging Face (7)
      • Pytorch Lightning (8)
      • Lang Chain (0)
    • Coding (35)
      • Python (25)
      • Numpy (5)
      • Clean Code (3)
      • SQL (1)
    • Algorithm (45)
      • BOJ (25)
      • 프로그래머스 (17)
    • Django (1)
    • Cloud (1)
      • GCP (1)
    • Program (7)
      • Docker (5)
      • VSCode (2)
    • AI API (3)
      • OpenAI (2)
      • Gemini (1)
    • Project (1)
    • 활동 (7)
      • 교내활동 (4)
      • 대외활동 (2)
      • 잡동사니 (1)

Tag

DataSet, numpy, Metric, c++, ESconv, 오블완, 실버, paper, LEVEL 2, 백준, Silver, Python, 파이썬, PyTorch Lightning, BOJ, huggingface, pytorch, 프로그래머스, nlp, docker,

최근글과 인기글

  • 최근글
  • 인기글

최근댓글

공지사항

페이스북 트위터 플러그인

  • Facebook
  • Twitter

Archives

Calendar

«   2025/05   »
일 월 화 수 목 금 토
1 2 3
4 5 6 7 8 9 10
11 12 13 14 15 16 17
18 19 20 21 22 23 24
25 26 27 28 29 30 31

방문자수Total

  • Today :
  • Yesterday :

Copyright © Kakao Corp. All rights reserved.

티스토리툴바