在Docker下使用Nvidia GPU進行計算

2021-08-07 13:20:52 字數 1150 閱讀 3900

要使用nvidia-docker來執行docker 容器

nvidia-docker run -p 8503:8503 --mount type=bind,source=sentiment-phone_pre/tf_model,target=/models/sentiment-analysis -e model_name=sentiment-analysis -t tensorflow/serving

nvidia-docker exec -it contain_id /bin/bash

【系統:centos 7, nvidia titan x】 

最近在使用gpu計算,同時也在使用docker做虛擬化環境。那麼問題來了,在虛擬機器下如何使用gpu呢?

在網上找到了很多答案,例如使用lxc等。

實際上將gpu對映到docker下很容易,只需要在啟動映象的時候將裝置掛載上去即可。

檢視顯示卡裝置名

[root@*** ~]# ls -la /dev | grep nvidia

crw-rw-rw-. 1 root root 195, 0 sep 1613:

49 nvidia0

crw-rw-rw-. 1 root root 195, 255 sep 1613:

49 nvidiactl

crw-rw-rw-. 1 root root 247, 0 sep 1613:

54 nvidia-uvm

docker

run-it-

-name

name-v

/home/:/mnt/home--

privileged=true--

device

/dev/nvidia

-uvm:/dev/nvidia

-uvm--

device

/dev/nvidia0:/dev/nvidia0--

device

/dev/nvidiactl:/dev/nvidiactl

myconda:cuda

bash

這裡注意需要使用 privilege 模式,每個裝置都用一次--device掛載

虛擬機器內安裝驅動以及cuda

在Docker下使用Nvidia GPU進行計算

系統 centos 7,nvidia titan x 最近在使用gpu計算,同時也在使用docker做虛擬化環境。那麼問題來了,在虛擬機器下如何使用gpu呢?在網上找到了很多答案,例如使用lxc等。實際上將gpu對映到docker下很容易,只需要在啟動映象的時候將裝置掛載上去即可。檢視顯示卡裝置名 ...

docker 在Windows下使用遇到的坑

1.大部分系統不支援直接安裝docker for windows,只能使用docker toolbox,相當於在windows上安裝了乙個linux的虛擬機器 3.直接使用docker run 執行ubuntu的映象時會出現 the input device is not a tty.if you ...

docker在Linux下的安裝與使用

本文主要介紹下在ububuntu下,docker的安裝。安裝成功後,可以檢視下docker版本 安裝完成後,可以獲取映象。第一步,需要到遠端的倉庫進行搜尋 找到想要安裝的映象對應的版本。執行命令,比如 docker pull redis 3.0 通過docker images檢視映象 配置完成後,獲...