Closed robinsdeepak closed 4 years ago
I guess it's because your cudatoolkit is 9. or 10.. If you use anaconda, you can simply:
conda install cudatoolkit==8.0
or some other version.
Or instead, you can use an updated tensorflow like conda install tensorflow-gpu==1.12.0rc2
.
@ecustWallace Thanks for the address I saw from requirements.txt that the source code requires tensorflow-gpu version 1.1.0 to run. Does tensorflow-gpu version 1.12 actually work for this source code?
Hey, I'm getting error while training the data after extraction, Can you help me with it?