因为想要搞一搞ML,实验室服务器排不上,只好自己搭个环境。
win10
1,IDE用的pycharm,在pycharm里面下载的TensorFlow-gpu 1.12.0;
2,然后查看显卡支持哪个版本的CUDA,控制面板 - >所有控制面板项 - >双击NVIDIA控制面板项

在NVIDIA面板控制里面选择通过预览调整图像设置 ,选择再 帮助里面的系统-信息

系统信息的组件里面就有显卡适配的CUDA版本。我的是CUDA10.0.150

3、去NVIDIA官网下载CUDA
4、下载CUDNN需要注册用户,不过很容易
注册后去找与你下载的CUDA版本对应的CUDNN版本

提示:用浏览器下载很慢的,可以把下载链接复制到迅雷里面,一会就下好了。
============================================================================
抱歉哈,没安上,cuda和tensorflow的版本不匹配,尝试了好多版本,终于,tensorflow-gpu1.8+cuda9.0.103+cudnn7.1,成功了。
本文详细介绍了在Windows 10环境下搭建TensorFlow-GPU深度学习环境的步骤,包括选择合适的CUDA和cuDNN版本,以及解决版本不匹配的问题。
1999

被折叠的 条评论
为什么被折叠?



