【已解决】30系显卡深度学习,jupyter notebook内核挂掉了,需要重启

tensorflow深度学习遇到各种疑难杂症,一律先考虑是不是工具包版本不匹配

安装正确对应版本的显卡驱动,python版本,cuda版本,cudnn版本,缺一不可,对应关系如下:

【已解决】30系显卡深度学习,jupyter notebook内核挂掉了,需要重启

最新版本的显卡驱动程序通常可以正常使用。

[En]

The latest version of the graphics card driver is generally fine.

我遇到了几个问题包括:

1、训练第一个epoch很慢,要加载好几分钟,因为我是30系卡,原生之支持cuda11及cudnn8以上,低版本需要兼容,效率很低。

2、训练时jupyter notebook内核挂掉。

3、训练时gpu占用率为0。

以上问题在安装新版tensoflow和正确版本的工具包后一次性全部消失,最后分享一下我现在的工具包配置:

RTX3050 + python3.8 64位 + NVidia 516.59版驱动 + tensorflow2.9.0 + CUDA11.2 + cudnn8.1

顺便说一句,每个工具包的官方网站都链接到了附录:

[En]

By the way, the official website of each toolkit is linked to the appendix:

cuda下载:CUDA Toolkit Archive | NVIDIA Developer

cudnn下载:cuDNN Archive | NVIDIA Developer

n卡驱动下载:官方 GeForce 驱动程序 | NVIDIA

python下载:Download Python | Python.org

Original: https://blog.csdn.net/weixin_46539334/article/details/125753848
Author: 杨熠楚
Title: 【已解决】30系显卡深度学习,jupyter notebook内核挂掉了,需要重启

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/514788/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球