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

Updates for build configurations. #126

Closed veritas9872 closed 1 year ago

veritas9872 commented 1 year ago

Make cuDNN v8 related build and runtime configurations explicit. Also enable CUDA lazy loading, which is effective for CUDA 11.7+ and especially for 11.8+. Change the structure of the Dockerfiles so that the USER is no longer specified inside the Dockerfile but at runtime by the Docker Compose file. This makes writing the Dockerfile much easier while preserving the same functionality.