| # A100 训练环境依赖 (Python 3.10/3.11, CUDA 12.x) | |
| torch>=2.1 | |
| torchvision>=0.16 | |
| diffusers>=0.27 | |
| accelerate>=0.28 | |
| omegaconf>=2.3 | |
| einops | |
| lpips | |
| pyiqa | |
| opencv-python | |
| scipy | |
| numpy | |
| pillow | |
| safetensors | |
| huggingface_hub | |
| pyyaml | |
| requests | |
| # A100 训练环境依赖 (Python 3.10/3.11, CUDA 12.x) | |
| torch>=2.1 | |
| torchvision>=0.16 | |
| diffusers>=0.27 | |
| accelerate>=0.28 | |
| omegaconf>=2.3 | |
| einops | |
| lpips | |
| pyiqa | |
| opencv-python | |
| scipy | |
| numpy | |
| pillow | |
| safetensors | |
| huggingface_hub | |
| pyyaml | |
| requests | |