2024-03-30 20:54:23 +00:00
|
|
|
apiVersion: v1
|
|
|
|
kind: Pod
|
|
|
|
metadata:
|
|
|
|
name: cuda-test
|
2024-03-30 20:55:59 +00:00
|
|
|
namespace: default
|
2024-03-30 20:54:23 +00:00
|
|
|
spec:
|
|
|
|
containers:
|
|
|
|
- name: cuda-container
|
2024-03-31 01:42:46 +00:00
|
|
|
image: nvidia/cuda:11.0.3-base-ubuntu20.04
|
2024-03-30 20:54:23 +00:00
|
|
|
command: ["nvidia-smi"]
|
|
|
|
resources:
|
|
|
|
limits:
|
|
|
|
nvidia.com/gpu: 1
|