forked from Ceruleanacg/Personae
-
Notifications
You must be signed in to change notification settings - Fork 0
/
docker-playbook.yml
54 lines (53 loc) · 1.83 KB
/
docker-playbook.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
- hosts: gpus
remote_user: shuyu
become: yes
become_method: su
tasks:
- name: Update Package Part-1
apt:
update_cache: yes
- name: Install Docker Dependency Part-1
apt:
name: apt-transport-https
state: latest
- name: Install Docker Dependency Part-2
apt:
name: ca-certificates
state: latest
- name: Install Docker Dependency Part-3
apt:
name: curl
state: latest
- name: Install Docker Dependency Part-4
apt:
name: software-properties-common
state: latest
- name: Add Docker GPG Key
shell: curl -fsSL https://download.docker.com/linux/ubuntu/gpg | apt-key add -
- name: Add Repo
apt_repository:
repo: deb [arch=amd64] https://download.docker.com/linux/ubuntu xenial stable
state: present
- name: Install Docker
apt:
name: docker-ce
update_cache: yes
- name: Add Nvidia GPG Key Part-1
shell: curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
- name: Add Nvidia GPG Key Part-2
shell: curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
- name: Install Nvida-Docker
apt:
name: nvidia-docker2
update_cache: yes
- name: Add Docker User
raw: useradd -p $(openssl passwd -1 duser@hflab) duser
- name: Append Docker User Group
raw: usermod -aG docker duser
- name: Change Nvidia Docker Runtime
raw: sed '2i \ \ \ \ "default-runtime":"nvidia",' /etc/docker/daemon.json
- name: Restart Docker
raw: service docker restart
- name: Run Nvida-Docker Nvidia-smi
shell: docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
# FIRST: ansible-console gpus -k -K -b --become-method=su