Nvidia Docker安裝說明(含WSL2)

4027 단어 wsl2docker
  • 訂閱patreon即可看到更多文章https://www.patreon.com/wade3c
  • Youtube頻道



  • Linux에서 docker를 다운로드할 수 있습니다.

    https://docs.docker.com/engine/install/ubuntu/



    curl -fsSL https://get.docker.com -o get-docker.sh
    



    sudo sh get-docker.sh
    


    WSL2




  • > WSL2는 다양한 최신 버전과 Linux 통합 솔루션
  • > 21H2 後也不用去安裝特殊版的Nvidia driver了
  • > docker desktop에서 3.1 버전의 WSL2에서 nvidia-docker로 업그레이드

  • WSL 내부의 Docker는 Linux에서 사용할 수 있고 시간이 오래 걸리는 회의를 소집할 수 있습니다.

    리눅스



    要先把nvidia의 패키지 서버 給加進apt의 소스 목록

    distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
          && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
          && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
                sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
                sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
    


    然後就可以直接去裝nvidia-docker套件了

    sudo apt-get update
    sudo apt-get install -y nvidia-docker2
    


    裝好後記得要去把舊有的docker服務給關掉

    sudo service docker restart
    


    用Ubuntu 話就不用去設定 /etc/docker/daemon.json
    "default-runtime":"nvidia","runtimes":{"nvidia":{"path":"nvidia-container-runtime","runtimeArgs":[]}}
    


    테스트



    有多種image可以進行測試,挑一個自己喜歡的就行

    最簡單的就是去看nvidia-smi能不能順利運作

    sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
    


    참조



    Installation Guide — NVIDIA Cloud Native Technologies documentation

    WSL 2 GPU Support for Docker Desktop on NVIDIA GPUs - Docker

    좋은 웹페이지 즐겨찾기