Open saturn118 opened 5 years ago
I ran the code on my Mint 19.1 with Cuda 8.0, GCC < 5 and compiled the Flownet with sm_61 for my Nvidia GTX1070Ti (Pascal).
I am able to sucessfully run the code using cuda 9
torch 0.4.0 cuda 9.0 python3.6 work.
OS??
Hi, can you specify which CUDA toolkit you installed (8,9,10) and the operating system that you used. I'm having constant issues trying to get the demo running and I think it's due to the wrong cuda version for pytorch 0.4.0