zhhlee/InterFusion

Trying to run it on a cpu.

Ashish-Tiger opened this issue · 1 comments

Hello,
Is it possible to run this on CPU?
I tried to run it on Colab but seems like this code is dependent on CUDA 9 and Colab doesn't support it.
Can you point me, how I can try this on a CPU or maybe Colab?

The issue was solved by skipping installation of GPU TensorFlow and then manually getting the wheel for the same version of CPU and installing the required library from the wheel file.