16.03.2025, 15:06
vs-deoldify with CUDA 12.8
I was able to get vs-deoldify working with CUDA 12.8 as explained in post: #11
To get "colormnet" working is necessary to recompile "Pytorch-Correlation-extension".
To get "Pytorch-Correlation-extension" working is necessary to set the following environment variable (where has been added the compute capability of RTX 50 series: 12.0)
I attached the updated instruction for compiling Pytorch-Correlation-extension with CUDA 12.8 and the working wheel build.
Dan
P.S.
see: matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards
I was able to get vs-deoldify working with CUDA 12.8 as explained in post: #11
To get "colormnet" working is necessary to recompile "Pytorch-Correlation-extension".
To get "Pytorch-Correlation-extension" working is necessary to set the following environment variable (where has been added the compute capability of RTX 50 series: 12.0)
TORCH_CUDA_ARCH_LIST=8.0 8.6 9.0 12.0
I attached the updated instruction for compiling Pytorch-Correlation-extension with CUDA 12.8 and the working wheel build.
Dan
P.S.
see: matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards