目 录
1 准备工作
Win10上搭建TensorFlow的开发环境需要至少需要安装3个软件,分别为:1.python,2.CUDA和CuDNN,3.TensorFlow(GPU版)。安装前一定要确认好每个软件的版本是否相互支持。
1.1(坑一)确认电脑上没有安装tensorflow的CPU版本,有的话要先卸载
1.2(坑二)更新自己的显卡驱动
因为刚刚重装了电脑发现显卡驱动竟然丢了,那还怎么用GPU,赶紧到官网下载最新的,否则会报错:cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
也可以在命令行运行命令nvidia-smi查看自己的显卡状态是否有问题
1.3 查看python版本
import sys
sys.version
‘3.7.3 (v3.7.3:ef4ec6ed12, Mar 25 2019, 22:22:05) [MSC v.1916 64 bit (AMD64)]’
1.4 查看GPU版本
我的电脑 -> 属性 -> 设备管理器 -> 显示适配器,我的版本是NVIDIA GeForce GTX 1050 with Max-Q Design
1.5 查看Tensorflow版本
GitHub:https://github.com/tensorflow/tensorflow/releases
Pypi:https://pypi.org/project/tensorflow/#history
1.6 查看版本支持
windows和VS的CUDA支持:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
可以看到CUDA10.1(因为有网友回复10.1会有冲突,而且我后面安装tf2.0的时候是不支持10.1的,所以这里先安装10.0吧)支持win10和VS2019的版本
目前官网上没有给出py3.7和tensorflow1.13的支持,但是有看到网友成功安装的帖子,所以这里进行试装
GPU的CUDA支持:https://developer.nvidia.com/cuda-gpus
显示支持6.1以上版本
2 安装CUDA Toolkit 10.0
安装10.0吧,虽然我自己试过10.1可以安装,但是很多人还是不行
下载地址:https://developer.nvidia.com/cuda-toolkit-archive
运行cuda_10.1.105_418.96_win10.exe,修改安装路径,之后会进行兼容性检查(按照上面对应的版本应该是没有问题的),同意许可协议后选择自定义安装