Open sdspieg opened 1 year ago
This library uses PyTorch under the hood rather than Tensorflow, but other than that whatever version of CUDA will work with the version of PyTorch you want to install should be fine. It looks like the most recent version of PyTorch (1.13.1) works with CUDA 11.6 or 11.7 (https://pytorch.org/get-started/locally/#start-locally).
Just sharing progress (from a Windows machine with a GPU)
That SEEMS to work - I see that my gpu is using 6GB of VRAM. Not sure about time improvement, though...
Any suggestions for which versions of Tensorflow, CUDA and cuDNN to run?