從零開始的docker容器深度學習環境配置

這篇博客已經介紹了關於如何在計算節點配置使用docker容器,但是直接下載的nvidia/cuda容器,除了cudnn和cuda,其他必須的工具都沒有,本文將重點記錄如何在docker容器中配置深度學習的開發環境

1 配置apt-get

使用

apt-get update

命令更新apt-get的軟件列表

2 安裝配置vim

2.1 安裝vim

apt-get install vim

2.2 配置vim

新建或打開vim的配置文件

vim ~/.vimrc

輸入配置信息

set number # 在vim中顯示行號
:set tabstop=4 # 將vim的tab鍵替換成4個空格

3 安裝配置python

3.1 安裝python和pip

安裝python

apt-get install python3.5

安裝pip

apt-get install python3-pip

3.2 配置pip

給pip換源

mkdir ~/.pip

然後添加如下配置信息,以清華源爲例

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

更新pip

pip3 install --update pip # 以後使用pip3命令換成了pip

4 安裝opencv

opencv需要很多第三方依賴,需要安裝許多額外的庫
安裝python

pip install opencv-python

importopencv的使用可能會遇到如下格式的報錯
在這裏插入圖片描述

ImportError: libgthread-2.0.so.0: cannot open shared object file: No such file or directory

解決方案是使用apt-get安裝對應的依賴

apt-get install libglib2.0-dev

類似的,博主在配置環境的時候還被要求安裝下列依賴:

ImportError: libSM.so.6: cannot open shared object file: No such file or directory
apt-get install libsm6
ImportError: libXrender.so.1: cannot open shared object file: No such file or directory
apt-get install libxrender1

一種更通用的解決方案是使用apt-get的搜索功能,之後會提示需要安裝的依賴

apt-file search libSM.so.6

5 安裝tensorflow

安裝需要注意cuda、cudnn、python、tensorflow的版本號,如下圖所示
在這裏插入圖片描述
在這裏插入圖片描述
本文以cuda版本9,cuDNN版本7,python版本3.5爲例

pip install tensorflow-gpu==1.11.0

6 安裝配置git

6.1 安裝git

apt-get install get

6.2 配置git

superGG1990有更詳細的博文

配置用戶名和郵箱
git config --global user.name "NAME"
git config --global user.email "[email protected]"
生成並配置密鑰

生成密鑰

ssh-keygen -t rsa -C "[email protected]"

查看密鑰,ls以後id_dsa是私鑰,id_dsa.pub是共鑰

cd ~/.ssh

需要將公鑰提交的github之類的代碼託管平臺上,
登陸github帳戶,點擊頭像,然後 Settings -> 左欄點擊 SSH and GPG keys -> 點擊 New SSH key
id_sda.pub裏的內容複製進去
最後在容器內驗證是否添加成功

ssh -T [email protected]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章