cresset-template / cresset

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.
MIT License
715 stars 41 forks source link

Minor build argument fix and addition of documentation on variable priority order. #131

Closed veritas9872 closed 1 year ago

veritas9872 commented 1 year ago

Remove cuDNNv8 experimental flag as it is enabled on PyTorch 2.x anyway. Also, I do not like anything labeled "experimental" on the main image. Users can add that themselves.

Also remove confusing defaults from the Dockerfile and add many comments on how arguments are passed on from the .env file to the docker-compose.yaml file and finally the Dockerfile.