win10戴尔游侠GTX1050TI+TensorFlow-gpu+CUDA10.0.130+CUDNN7.4.1.5配置深度学习环境

因为想要搞一搞ML,实验室服务器排不上,只好自己搭个环境。

win10
1,IDE用的pycharm,在pycharm里面下载的TensorFlow-gpu 1.12.0;

2,然后查看显卡支持哪个版本的CUDA,控制面板 - >所有控制面板项 - >双击NVIDIA控制面板项

   

         win10戴尔游侠GTX1050TI+TensorFlow-gpu+CUDA10.0.130+CUDNN7.4.1.5配置深度学习环境_第1张图片

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

win10戴尔游侠GTX1050TI+TensorFlow-gpu+CUDA10.0.130+CUDNN7.4.1.5配置深度学习环境_第2张图片

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

win10戴尔游侠GTX1050TI+TensorFlow-gpu+CUDA10.0.130+CUDNN7.4.1.5配置深度学习环境_第3张图片

3、去NVIDIA官网下载CUDA

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

4、下载CUDNN需要注册用户,不过很容易

注册后去找与你下载的CUDA版本对应的CUDNN版本

win10戴尔游侠GTX1050TI+TensorFlow-gpu+CUDA10.0.130+CUDNN7.4.1.5配置深度学习环境_第4张图片

提示:用浏览器下载很慢的,可以把下载链接复制到迅雷里面,一会就下好了。

============================================================================

抱歉哈,没安上,cuda和tensorflow的版本不匹配,尝试了好多版本,终于,tensorflow-gpu1.8+cuda9.0.103+cudnn7.1,成功了。

 

你可能感兴趣的:(机器学习)