apiVersion: v1
kind: Pod
metadata:
  name: theano
  labels:
    purpose: theano
spec:
  containers:
    - name: theano-container
      image: jammingyu9/ai:theano
      workingDir: /root
      env:
      - name: HIP_VISIBLE_DEVICES
        value: "0,1" # # 0,1,2,...,n for running on GPU and select the GPUs, -1 for running on CPU 
      command: ["/bin/bash", "-c", "--"]
      args: ["sleep infinity & wait"]
      resources:
        limits:
          amd.com/gpu: 2 # requesting a GPU
