24 lines
440 B
YAML
24 lines
440 B
YAML
---
|
|
# Debian/Ubuntu-specific tasks
|
|
|
|
- name: Install dependencies
|
|
apt:
|
|
name:
|
|
- build-essential
|
|
- cmake
|
|
- git
|
|
- python3
|
|
- python3-pip
|
|
state: present
|
|
update_cache: yes
|
|
|
|
- name: Check for NVIDIA GPU
|
|
shell: lspci | grep -i nvidia
|
|
register: nvidia_check
|
|
failed_when: false
|
|
changed_when: false
|
|
|
|
- name: Set GPU type
|
|
set_fact:
|
|
gpu_type: "{{ 'nvidia' if nvidia_check.rc == 0 else 'none' }}"
|