在Docker下使用Nvidia GPU進行計算

2021-07-05 08:04:21 字數 952 閱讀 8186

【系統: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

sep1613:

49 nvidia0

crw-rw-rw-. 1 root root 195, 255

sep1613:

49 nvidiactl

crw-rw-rw-. 1 root root 247, 0

sep1613:

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進行計算

要使用nvidia docker來執行docker 容器 nvidia docker run p 8503 8503 mount type bind,source sentiment phone pre tf model,target models sentiment analysis e mode...

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檢視映象 配置完成後,獲...