安装nvidia-docker

安装nvidia-docker方能在docker底下調用gpu資源使用方式如github中記載.

以下為 Ubuntu 安裝方法

# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb

# Test nvidia-smi
sudo nvidia-docker run --rm nvidia/cuda nvidia-smi

經測試在若出現錯誤無法安裝,可能是相關套件更新後導致異常,可參考（連結）.

安裝完畢後可以下載tensorflow版本之image執行並測試.

tensorflow-cpu 版本

nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest

tensorflow-gpu 版本

nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-gpu

執行container後會啟動jupyter界面可以直接撰寫並測試tensorflow.

並且在環境中已有幾個範例提供學習.

 

 

 

arrow
arrow
    文章標籤
    tensorflow Docker
    全站熱搜

    Lung-Yu,Tsai 發表在 痞客邦 留言(0) 人氣()

    E-mail轉寄