Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Can I train this model on RTX 3090 GPU #47

Open
lzhzju opened this issue Aug 9, 2022 · 2 comments
Open

Can I train this model on RTX 3090 GPU #47

lzhzju opened this issue Aug 9, 2022 · 2 comments

Comments

@lzhzju
Copy link

lzhzju commented Aug 9, 2022

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
Copy link

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

@WenBingo
Copy link

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@yifeipet @lzhzju @WenBingo