How to use GPU in docker container
前言
基本上原生的docker 容器是無法使用GPU資源解決方法
必須使用NVIDIA的 Docker才可以讓容器吃到GPU的資源以下是在Ubuntu上的安裝步驟
Step 1. (Optional) 安裝驅動
如果GPU的驅動已經裝好了化可以先略過這一步sudo ubuntu-drivers autoinstall
安裝完之後可以用以下指令確認有沒有裝成功
nvidia-smi
Step 2. 安裝 CUDA
CUDA是個運算架構能夠輕易地時線平行運算
wget https://gist.githubusercontent.com/acerkr71a/29ecd032eeb178888fec0bb7660cc3bf/raw/03df656e945d7c56ee2d454e0ea83796e8eacfeb/install-cuda.sh && chmod +x installer.sh && ./installer.sh
Step 3. 安裝NVIDIA 的 Docker
wget https://gist.githubusercontent.com/acerkr71a/7c19e940f4029d86476309d63950d750/raw/a7f368c800b4ef2e23559ef5f13f97a540dc4499/install-nvidia-docker.sh && chmod +x docker-installer.sh && ./docker-installer.sh
留言
張貼留言