diff --git a/.woodpecker.yml b/.woodpecker.yml index 9b6e843..2e2ea6c 100644 --- a/.woodpecker.yml +++ b/.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: