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.
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 thedocker-compose.yaml
file and finally theDockerfile
.