TensorFlow的环境配置与安装教程详解(win10+GeForce GTX1060+CUDA 9.0+c

2020-06-22 13:16 来源:易采站长站 作者:王冬梅 点击: 评论:

A-A+

原标题:TensorFlow的环境配置与安装教程详解(win10+GeForce GTX1060+CUDA 9.0+c

记录一下安装win10+GeForce GTX1060+CUDA 9.0+cuDNN7.3+tensorflow-gpu 1.12.0+python3.5.5

之前已经安装过pycharm、Anaconda以及VS2013,因此,安装记录从此后开始

总体步骤大致如下:

1、确认自己电脑显卡型号是否支持CUDA(此处有坑)

此处有坑!不要管NVIDIA控制面板组件中显示的是CUDA9.2.148。

你下载的CUDA不一定需要匹配,尤其是CUDA9.2,最好使用CUDA9.0,我就在此坑摔的比较惨。

2、下载CUDA以及cuDNN,注意版本对应①查看版本匹配:

https://www.tensorflow.org/install/source_windows

②下载CUDA:

https://developer.nvidia.com/cuda-toolkit-archive

官网上下载的CUDA 9.0有好几个版本,其中主要是cuda_9.0.176_win10.exe,其他的四个是补丁。

③下载cuDNN:

https://developer.nvidia.com/cudnn

https://developer.nvidia.com/rdp/cudnn-archive

下载cuDNN需要注册一个NVIDIA的账号。

3、安装CUDA和cuDNN,并设置环境变量(重要)①CUDA安装

我是按照默认路径安装的,没有修改。此外,使用自定义安装,但是几乎全选了,除了一个当前版本已经是最新版本的组件没有勾选。

切记CUDA的安装路径,因为安装cuDNN以及设置环境变量时需要。

②cuDNN9.0安装

cuDNN是一个压缩包,解压后的内容如下

全选并复制所有内容,粘贴到CUDA的安装路径下,默认路径是:

③设置环境变量(重要)

这部分我主要参考的是:https://blog.csdn.net/qilixuening/article/details/77503631

计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了两个环境变量,接下来,分别是:

CUDA_PATH和CUDA_PATH_V8_0。

还要在系统变量中新建以下几个环境变量:

【易采站长站编辑:秋军】