nouhadziri / THRED

The implementation of the paper "Augmenting Neural Response Generation with Context-Aware Topical Attention"
https://arxiv.org/abs/1811.01063
MIT License
111 stars 25 forks source link

[UPDATE] tensorflow_gpu version #39

Closed jacksonchen1998 closed 4 months ago

jacksonchen1998 commented 4 months ago

This can be reproduced by changing the tensorflow_gpu from 1.12.2 to 1.15.0.

My GPU version GeForce RTX 4090. CUDA version 12.0. Python version 3.7.

I choose to use pip to create my environment not using conda.

ehsk commented 4 months ago

Thanks for the update!