diff --git a/Dockerfile b/Dockerfile index 6c9bd35..540c892 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -ARG CUDA_VERSION=12.1.0-devel +ARG CUDA_VERSION=12.4.0-devel FROM --platform=amd64 docker.io/nvidia/cuda:${CUDA_VERSION}-ubuntu22.04 diff --git a/README.md b/README.md index 113a395..513a035 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,7 @@ This is an extensive node suite that enables ComfyUI to process 3D inputs (Mesh - [Pre-builds](https://github.com/MrForExample/Comfy3D_Pre_Builds) are available for: - Windows 10/11, Ubuntu 22.04 - Python 3.10/3.11/3.12 - - CUDA 12.1/11.8 + - CUDA 12.4/12.1/11.8 - torch 2.3.0+cu121/cu118, torchvision 0.18.0+cu121/cu118 - [install.py](install.py) will download & install Pre-builds automatically according to your runtime environment, if it couldn't find corresponding Pre-builds, then [build script](_Pre_Builds/_Build_Scripts/auto_build_all.py) will start automatically, if automatic build doesn't work for you, then please check out [Semi-Automatic Build Guide](_Pre_Builds/README.md#build-required-packages-semi-automatically) - If you have any missing node in any open Comfy3D workflow, try simply click [Install Missing Custom Nodes](https://github.com/ltdrdata/ComfyUI-Manager?tab=readme-ov-file#support-of-missing-nodes-installation) in ComfyUI-Manager diff --git a/_Pre_Builds/_Build_Scripts/build_config.yaml b/_Pre_Builds/_Build_Scripts/build_config.yaml index 550f536..6002787 100644 --- a/_Pre_Builds/_Build_Scripts/build_config.yaml +++ b/_Pre_Builds/_Build_Scripts/build_config.yaml @@ -1,6 +1,6 @@ repo_id: "MrForExample/Comfy3D_Pre_Builds" -supported_cuda_versions: ["12.1", "11.8"] -cuda_version: "12.1" +supported_cuda_versions: ["12.4", "12.1", "11.8"] +cuda_version: "12.4" # Packages inside this file will be pip downloaded before actual build start build_requirements: "build_requirements.txt" @@ -17,17 +17,18 @@ build_base_packages: ["torch", "torchvision"] # All packages will be automatically installed at beginning of install.py follwing orders from top to bottom remote_packages: torch: - version: "2.4.0" + version: "2.5.1" url: "https://download.pytorch.org/whl/${cuda_version}" torchvision: - version: "0.19.0" + version: "0.20.1" url: "https://download.pytorch.org/whl/${cuda_version}" - torch-scatter: - url_option: "--find-links" - url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html" - xformers: - version: "0.0.27.post2" - url: "https://download.pytorch.org/whl/${cuda_version}" \ No newline at end of file + version: "0.0.28.post3" + url: "https://download.pytorch.org/whl/${cuda_version}" + + # At the time of writing, torch-scatter is not available for PyTorch 2.5.0 + #torch-scatter: + # url_option: "--find-links" + # url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html" \ No newline at end of file diff --git a/_Pre_Builds/_Build_Scripts/dependencies.txt b/_Pre_Builds/_Build_Scripts/dependencies.txt index f0d3b74..a7c08db 100644 --- a/_Pre_Builds/_Build_Scripts/dependencies.txt +++ b/_Pre_Builds/_Build_Scripts/dependencies.txt @@ -1,4 +1,5 @@ https://github.com/ashawkey/diff-gaussian-rasterization.git https://github.com/NVlabs/nvdiffrast.git https://github.com/ashawkey/kiuikit.git -https://github.com/facebookresearch/pytorch3d.git \ No newline at end of file +https://github.com/facebookresearch/pytorch3d.git +https://github.com/rusty1s/pytorch_scatter.git \ No newline at end of file