1、安装 tensorflow-gpu,keras,需要提前知晓适显卡配置,根据显卡配置下载cudatoolkit,cudnn两个模块。(途中遇到更换jupyter notebook内核的问题)
第一步:查看显卡配置
1、打开控制面板查看信息。2、帮助-->系统信息-->组件。3、可以看到版本,本机cuda版本为11.6,后面选择的cuda版本不要超过这里的版本就好
4、nvidia-smi
注意pytorch和cuda版本对应 CUDA Deep Neural Network (cuDNN) | NVIDIA Developer 选择合适的cudnn 版本安装
cuDNN驱动下载:cuDNN Archive | NVIDIA Developer
5、
Previous