diff --git a/.gitea/workflows/ci.yml b/.gitea/workflows/ci.yml index 6aae617..0daf2ec 100644 --- a/.gitea/workflows/ci.yml +++ b/.gitea/workflows/ci.yml @@ -48,6 +48,7 @@ jobs: image: ${{ vars.CONTAINER_REGISTRY }}/${{ env.IMAGE_NAME }}:latest options: >- --gpus all + --privileged --env NVIDIA_DRIVER_CAPABILITIES=all --env NVIDIA_VISIBLE_DEVICES=all steps: