Torch device cuda is available. Linux как проверить версию cuda.

« Предыдущее