因为想要搞一搞ML,实验室服务器排不上,只好自己搭个环境。
win10
1,IDE用的pycharm,在pycharm里面下载的TensorFlow-gpu 1.12.0;
2,然后查看显卡支持哪个版本的CUDA,控制面板 - >所有控制面板项 - >双击NVIDIA控制面板项
在NVIDIA面板控制里面选择通过预览调整图像设置 ,选择再 帮助里面的系统-信息
系统信息的组件里面就有显卡适配的CUDA版本。我的是CUDA10.0.150
3、去NVIDIA官网下载CUDA
/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
4、下载CUDNN需要注册用户,不过很容易
注册后去找与你下载的CUDA版本对应的CUDNN版本
提示:用浏览器下载很慢的,可以把下载链接复制到迅雷里面,一会就下好了。
============================================================================
抱歉哈,没安上,cuda和tensorflow的版本不匹配,尝试了好多版本,终于,tensorflow-gpu1.8+cuda9.0.103+cudnn7.1,成功了。