ubuntu18.04配置nvidia docker和遠程連接ssh+遠程桌面連接(一)


ubuntu18.04配置nvidia docker和遠程連接ssh+遠程桌面連接(一)

本教程適用於想要在遠程服務器上配置docker圖形界面用於深度學習的用戶。

(一)ubuntu18.04配置nvidia docker

配置環境:Ubuntu 18.04.1 LTS

第一步:安裝顯卡驅動和CUDA

安裝顯卡驅動和CUDA的教程網上有很多。本教程采用的是如下安裝方式。

參考網址:https://blog.csdn.net/yjt1325/article/details/84641062

CUDA和CUDNN在均可在官網下載。

CUDA:https://developer.nvidia.com/cuda-downloads

CUDNN:https://developer.nvidia.com/cudnn

本教程最終安裝的顯卡驅動版本是415.25,安裝的CUDA版本是10.0,安裝的CUDNN版本是7.4.2.24

安裝完成后可以通過nvidia-smi命令檢驗是否安裝成功:

第二步:安裝docker

依次輸入以下指令:

1 sudo apt-get update 2 sudo apt-get install  apt-transport-https  ca-certificates curl  software-properties-common 3 curl -fsSL  https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
4 sudo add-apt-repository "deb [arch=amd64] https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu $(lsb_release -cs) stable"
5 sudo apt-get update 6 sudo apt-get install docker-ce

配置完成后,輸入指令systemctl status docker檢測是否成功安裝並開啟docker:

第三步:安裝nvidia-docker2

依次輸入以下指令:

1 curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
2       sudo apt-key add -
3 distribution=$(. /etc/os-release;echo $ID$VERSION_ID) 4 curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
5       sudo tee /etc/apt/sources.list.d/nvidia-docker.list 6 sudo apt-get update 7 sudo apt-get install -y nvidia-docker2 8 sudo systemctl daemon-reload 9 sudo systemctl restart docker

配置完成后,輸入指令:sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi檢測nvidia-docker2是否安裝成功。(不報錯即為安裝成功)。

 

至此,支持GPU加速的nvidia-docker2已經安裝成功。但要想使用docker還需要拉取鏡像,並配置鏡像以符合我們的需求。下一期將介紹如何拉取可用於深度學習的鏡像(deepo)並配置ssh用於遠程訪問。

 

參考教程:

安裝docker:https://www.jianshu.com/p/07e405c01880

安裝nvidia-docker2:https://blog.csdn.net/xiangxianghehe/article/details/81216424


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM