[도커] 윈도우에서 WSL2를 사용하여 도커 설치

1. WSL2 설치

Windows PowerShell 관리자 권한으로 실행

dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart

# 설치 확인
wsl -l -v

---

2. Docker Desktop 설치

https://docs.docker.com/desktop/install/windows-install/

Install Docker Desktop on Windows

---

3. CUDA Toolkit 설치

(3090Ti CUDA11.2 기준)

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-wsl-ubuntu-12-0-local_12.0.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-0-local_12.0.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

 

4. 도커 이미지 로드

name, 경로 등 본인 환경 및 도커 이미지에 맞게 입력하시면 됩니다.

 

1) 도커 이미지 설치 (tar.gz 위치한 경로에서)

docker load -i cimjuk_yolo.tar.gz

2) 리눅스-wls2 nvidia 환경 충돌 해결

docker run -d --name cimjuk --rm pknu:cimjuk_yolo sleep 60
docker exec -u 0 cimjuk rm -rf /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1
docker exec -u 0 cimjuk rm -rf /usr/lib/x86_64-linux-gnu/libcuda.so.1
docker commit cimjuk pknu:cimjuk2

3) 도커 이미지 실행 (컨테이너 생성)

docker run --gpus all --shm-size=16g --name=cimjuk_yolo -ti pknu:cimjuk2 /bin/bash

4) 소스코드 실행

cd /workspace/cimjuk/code
python test.py --task test --batch-size 16 --data data/cimjuk.yaml --weights cimjuk.pt --name cimjuk_test --save-txt --save-conf --verbose