Skip to content

Commit 7409d18

Browse files
committed
feat(dev): only setup CUDA container toolkit if missing
This commit ensures that the CUDA container toolkit is only installed enabled if not yet done.
1 parent 509872c commit 7409d18

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

scripts/ansible/plays/setup_comfystream.yaml

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
hosts: all
44
become: yes
55
vars:
6-
docker_image: "livepeer/comfystream:latest"
6+
docker_image: "livepeer/comfystream:0.0.3"
77
comfyui_username: "comfyadmin"
88
comfyui_password: "Ge}_^jn?*?5#/|FWx9I&K|S0$/)#au8"
99

@@ -141,21 +141,34 @@
141141
create: yes
142142
notify: restart caddy
143143
# Install, configure, and start ComfyUI with Comfystream
144+
- name: Check if NVIDIA Container Toolkit is installed
145+
command: dpkg -l | grep nvidia-container-toolkit
146+
register: nvidia_toolkit_installed
147+
ignore_errors: yes
148+
changed_when: false
149+
- name: Check if NVIDIA runtime is configured in Docker
150+
command: docker info | grep -i nvidia
151+
register: nvidia_runtime_configured
152+
ignore_errors: yes
153+
changed_when: false
144154
- name: Add NVIDIA Container Toolkit repository key
145155
apt_key:
146156
url: https://nvidia.github.io/libnvidia-container/gpgkey
147157
keyring: /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
148158
state: present
159+
when: nvidia_toolkit_installed.rc != 0
149160
- name: Add NVIDIA Container Toolkit repository
150161
get_url:
151162
url: https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list
152163
dest: /etc/apt/sources.list.d/nvidia-container-toolkit.list
153164
mode: "0644"
165+
when: nvidia_toolkit_installed.rc != 0
154166
- name: Ensure NVIDIA repository uses correct signing key
155167
replace:
156168
path: /etc/apt/sources.list.d/nvidia-container-toolkit.list
157169
regexp: "^deb https://"
158170
replace: "deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://"
171+
when: nvidia_toolkit_installed.rc != 0
159172
- name: Install NVIDIA Container Toolkit
160173
apt:
161174
name: nvidia-container-toolkit
@@ -165,15 +178,18 @@
165178
delay: 30
166179
register: nvidia_toolkit_result
167180
until: nvidia_toolkit_result is success
181+
when: nvidia_toolkit_installed.rc != 0
168182
- name: Configure Docker to use NVIDIA runtime
169183
command: nvidia-ctk runtime configure --runtime=docker
170184
register: nvidia_ctk_result
171185
changed_when: nvidia_ctk_result.rc == 0
186+
when: nvidia_runtime_configured.rc != 0
172187
- name: Restart Docker service
173188
systemd:
174189
name: docker
175190
state: restarted
176191
enabled: yes
192+
when: nvidia_runtime_configured.rc != 0
177193
# Install, configure, and start ComfyUI with Comfystream
178194
- name: Install community.docker collection
179195
command: ansible-galaxy collection install community.docker

0 commit comments

Comments
 (0)