search

senguptaumd / Background-Matting

Background Matting: The World is Your Green Screen
https://grail.cs.washington.edu/projects/background-matting/
4.78k stars 662 forks source link

TensorFlow version #2

Closed gruossomonica closed 4 years ago

gruossomonica commented 4 years ago

Hi, are you sure about the version of TensorFlow using CUDA 10.0? Tensorflow 1.4 requires CUDA 8. I think you meant version 1.14. I'm trying with 1.15 and now it works well.

senguptaumd commented 4 years ago

Hi, You are right. That was a typo. It will be tensorflow=1.14 with CUDA10.0. Thanks for pointing that out. Corrected it in the README.

Teravus commented 4 years ago

I could be wrong, but the readme may require two equal signs for the version. tensorflow==1.14