Migration To Python 3.12/CUDA 12.4 - Success!

[seatv]

I was able to migrate DM without much trouble to Python v3.12.7 and CUDA 12.4. Here are the steps I followed:

Make sure you are on the latest version of pip

py -m pip install --upgrade pip

Install the following in the order below. I removed numpy from dependencies as installing that first caused issues; scikit-image had numpy dependency and it will install numpy.

pip install scikit-image 

pip install opencv_python

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124  

pip install matplotlib 

pip install tensorboardX

The above commands installs the following versions of various packages:

scikit-image - [imageio-2.37.0 lazy-loader-0.4 networkx-3.4.2 numpy-2.2.2 packaging-24.2 pillow-11.1.0 scikit-image-0.25.0 scipy-1.15.1 tifffile-2025.1.10]

opencv_python [4.11.0.86]

pytorch - [MarkupSafe-2.1.5 filelock-3.13.1 fsspec-2024.2.0 jinja2-3.1.3 mpmath-1.3.0 setuptools-70.0.0 sympy-1.13.1 torch-2.5.1+cu124 torchaudio-2.5.1+cu124 torchvision-0.20.1+cu124 typing-extensions-4.9.0]

matplotlib - [contourpy-1.3.1 cycler-0.12.1 fonttools-4.55.3 kiwisolver-1.4.8 matplotlib-3.10.0 pyparsing-3.2.1 python-dateutil-2.9.0.post0 six-1.17.0]

tensorboardX - [ protobuf-5.29.3 tensorboardX-2.6.2.2]