search

Justin-Tan / generative-compression

TensorFlow Implementation of Generative Adversarial Networks for Extreme Learned Image Compression
MIT License
511 stars 108 forks source link

Can I train this model on RTX 3090 GPU #47

Open lzhzju opened 1 year ago

lzhzju commented 1 year ago

Hi Justin,

May I ask if I can train this model on RTX 3090 GPU? It seems that tensorflow1.8 doesn't work properly on RTX3090.

Sincerely, Zihang

yifeipet commented 1 year ago

你就把他的code转成Tensorflow 2就行了。RTX30 系列的显卡需要更高版本的CUDA，而更高版本的CUDA不支持Tensorflow 1.8。

WenBingo commented 12 months ago

你就把他的code转成Tensorflow 2就行了。RTX30 系列的显卡需要更高版本的CUDA，而更高版本的CUDA不支持Tensorflow 1.8。

想问下你有转好的tensorflow 2的代码吗。我这能力有限，报错太多修改不好了 1286310571@qq.com