gpus

.woodpecker.yml

@@ -4,19 +4,25 @@ pipeline:
     image: busybox
     commands:
       - echo "pick pick wood pickasdpaasdsdick ....asd.sd..kj......"
-  nvidia-test:
-    image: nvidia/cuda:11.6.2-base-ubuntu20.04
-    commands:
-      - nvidia-smi
-      - apt update -y && apt install git build-essential wget -y
-      - git clone https://github.com/wilicc/gpu-burn
-      - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/libcublas-dev_10.1.0.105-1_amd64.deb
-      - dpkg -i libcublas-dev_10.1.0.105-1_amd64.deb
-      - cd gpu-burn
-      - make
-      - ./gpu_burn 60
+  gpu-stress:
+    image: waggle/gpu-stress-test:latest
     environment:
       - NVIDIA_VISIBLE_DEVICES=all
+    commands:
+      - python3 stress.py -m 5
+  #nvidia-test:
+  #  image: nvidia/cuda:11.6.2-base-ubuntu20.04
+  #  commands:
+  #    - nvidia-smi
+  #    - apt update -y && apt install git build-essential wget -y
+  #    - git clone https://github.com/wilicc/gpu-burn
+  #    - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/libcublas-dev_10.1.0.105-1_amd64.deb
+  #    - dpkg -i libcublas-dev_10.1.0.105-1_amd64.deb
+  #    - cd gpu-burn
+  #    - make
+  #    - ./gpu_burn 60
+  #  environment:
+  #    - NVIDIA_VISIBLE_DEVICES=all
 #  2nd:
 #    image: oguzpastirmaci/gpu-burn:latest
 #    environment: