Skip to content

Release v0.6.1 (January 9, 2026)

Choose a tag to compare

@lecode-official lecode-official released this 09 Jan 14:29
· 2 commits to main since this release

In order to allow users to choose a specific CUDA version when pulling the image, the Dockerfile was modified to also allow the specification of the PyTorch, CUDA, and cuDNN versions as build arguments. The GitHub Actions workflow was updated to build images for all available combinations of CUDA and cuDNN for the version of PyTorch used in the Dockerfile. The read me was updated to reflect this change.