DEPRECATION NOTICE! [Docker Image]

[fahadshery]

Hi,

I just came across this repo and it looks brilliant for fine tuning!

I am using docker and was able to build and run the image:

docker run -i -t --name ai-toolkit --gpus all ai-toolkit /bin/bash

I am getting a deprecation notice:

*************************
** DEPRECATION NOTICE! **
*************************
THIS IMAGE IS DEPRECATED and is scheduled for DELETION.
    https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/support-policy.md

 ______                 ______            _
(_____ \               (_____ \          | |
 _____) ) _   _  ____   _____) )___    __| |
|  __  / | | | ||  _ \ |  ____// _ \  / _  |
| |  \ \ | |_| || | | || |    | |_| |( (_| |
|_|   |_||____/ |_| |_||_|     \___/  \____|

For detailed documentation and guides, please visit:
https://docs.runpod.io/ and https://blog.runpod.io/

BTW, still using cuda 12.2 drivers so it's working fine btw! thought to let you know.

Other things:
the image size is about 20GB

[fahadshery]

the deprecation warning has disappeared now.
but the image size is still too big i.e. 20GB!

thanks