這篇博客已經介紹了關於如何在計算節點配置使用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
在import
opencv的使用可能會遇到如下格式的報錯
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]