1. 程式人生 > >構建自己的映象

構建自己的映象

參考連結https://www.cnblogs.com/Walther/p/7440534.html

tensorflow/tensorflow僅支援py2.7所以需要構建自己的基於py3.5的tensorflow映象

下載系統基礎映象,建立容器,在容器中安裝自己需要的軟體,python3,和cuda和cudnn對應的tensorflow-gpu,然後提交映象

1查詢ubuntu映象,建立容器 

nvidia-docker search ubuntu 

nvidia-docker run -it  -d -name myimage  ubuntu  bash

nvidia-docker exec  -it myimage bash 

在容器裡面安裝所需的py3.5 tensorflow

apt update
 apt install -y python3-pip python3-dev vim
 pip3 install tensorflow-gpu 

將安裝好的容器提交到映象 nvidia-docker commit myiamge 

方法二

構建自己的dockfile檔案

不熟悉