generative-compression icon indicating copy to clipboard operation
generative-compression copied to clipboard

Published 20 hours ago • verified publisher icon Justin-Tan

Can I train this model on RTX 3090 GPU

Open lzhzju opened this issue 2 years ago • 2 comments

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

lzhzju avatar Aug 09 '22 08:08 lzhzju

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

yifeipet avatar Aug 31 '22 21:08 yifeipet

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

想问下你有转好的tensorflow 2的代码吗。我这能力有限，报错太多修改不好了 [email protected]

WenBingo avatar Jul 20 '23 07:07 WenBingo