显卡驱动版本,CUDA版本,CUDA-tool版本,CuDnn版本。
直接讲实操步骤,
首先利用 cmd,输入 nvidia-smi
这里可以看到驱动版本
CUDA版本
那么我们去官网搜索tensorflow-gpu版本对应,会搜索到这样的表。在官网都有
那么我们需要装CUDA-tool,和cuDNN,我们刚才查到我们的CUDA版本是12.4,那么上面图代表的是CUDA-TOOL的版本,只要小于CUDA的版本都可以安装。而CUDA的版本是由显卡驱动决定的,如果我们想升级或降级自身的CUDA,就更新显卡驱动版本。即
选定CUDA-TOOL版本后,即可下载对应的CuDnn版本即可。
这样就完成了版本的对应。
tensorflow-gpu版本如何安装合适的版本呢?
于 2024-08-10 22:08:30 首次发布