cuda drivers

Author: rafaelpierrehf

.github/workflows/unit-test.yaml

@@ -36,7 +36,7 @@ jobs:
         wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb
         sudo dpkg -i cuda-keyring_1.1-1_all.deb
         sudo apt-get update
-        sudo apt-get -y install cuda-toolkit-12-3 nvidia-kernel-open-545 cuda-drivers-545
+        sudo apt-get -y install cuda-toolkit-12-3 cuda-drivers
     - name: nvidia-smi
       run: nvidia-smi
     - uses: actions/[email protected]

tox.ini

@@ -27,7 +27,7 @@ commands =
     pytest -s -v \
     {tty:--color=yes} \
     tests/unit/ {posargs} \
-    --log-cli-level=DEBUG \
+    --log-cli-level=ERROR \
     --log-format='%(asctime)s %(levelname)s %(module)s:%(lineno)d %(message)s'
 
 [testenv:unit-torch-slow]