fix: update dep + fix

[pommedeterresautee]

fix #264

• fix CUDA graph API
• update PyTorch dependency
• update Docker base image

local (bare metal) execution of pytest -x

=============================== warnings summary ===============================
test/test_model_optimization.py: 1 warning
test/test_torchdynamo.py: 79 warnings
  /home/geantvert/.local/share/virtualenvs/kernl/lib/python3.9/site-packages/torch/cuda/graphs.py:82: UserWarning: The CUDA Graph is empty. This ususally means that the graph was attempted to be captured on wrong device or stream. (Triggered internally at ../aten/src/ATen/cuda/CUDAGraph.cpp:191.)
    super(CUDAGraph, self).capture_end()

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
================ 2859 passed, 80 warnings in 8538.09s (2:22:18) ================

(updated image) Docker execution of pytest -x

=============================== warnings summary ===============================
test/test_model_optimization.py: 1 warning
test/test_torchdynamo.py: 79 warnings
  /usr/local/lib/python3.9/dist-packages/torch/cuda/graphs.py:82: UserWarning: The CUDA Graph is empty. This ususally means that the graph was attempted to be captured on wrong device or stream. (Triggered internally at ../aten/src/ATen/cuda/CUDAGraph.cpp:191.)
    super(CUDAGraph, self).capture_end()

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
================ 2859 passed, 80 warnings in 9597.66s (2:39:57) ==============