版本:
Tensorflow 2.1.0 + CUDA 10.1
【参考链接】https://jingyan.baidu.com/article/6fb756ec4fabc4241858fbf7.html
注意: 查询到的版本为对应的最高版本,可以低于该版本。如果已经下载tensorflow,则需要和tensorflow版本对应。【2.1.0以上版本的tensorflow没有经过特别指定的话,一般会自动下载GPU和CPU版本】
【官方CUDA和tensorflow对照表】
https://tensorflow.google.cn/install/source_windows#gpu
这里以10.1为例:
该路径为临时解压路径,软件安装好后会自动删除。【不建议更改路径】
解压过程中,如果电脑有安装360等杀毒软件,会弹出警告,点击允许所有操作即可。
解压完成后会自动启动,检查兼容性
许可协议
这里我们选择【自定义】
这里我们只选择第一个
如果没有安装VS C++编译器,把这个项目取消掉就可以了。这里我安装过VS2017,就不做演示了。
这里我们更改安装位置,当然也可以选择不更改。
点击【下一步】等待安装结束
同样的,如果电脑有安装360,在完成安装操作时会有预警弹窗,允许操作即可。
安装成功
在官网上找到对应版本下载解压即可。
解压后包含 3个文件夹和一个README
复制3个文件夹到之前安装CUDA时选择的路径,这里我的路径是D:\NVIDIA\CUDA
1. 添加系统变量
计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V10_0两个环境变量。
接下来,还要在系统中添加以下几个环境变量:
【在系统变量中点击新建】
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0(这是默认安装位置的路径,我的路径为D:\NVIDIA\CUDA Samples)
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
2. 编辑系统变量中的PATH
选中,编辑,然后添加如下路径:
%CUDA_LIB_PATH%
%CUDA_BIN_PATH%
%CUDA_SDK_LIB_PATH%
%CUDA_SDK_BIN_PATH%
【如果CUDA选择默认路径,则添加如下几条路径】
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64
【如果CUDA选择自定义路径,则参考我的路径】
D:\NVIDIA\CUDA\lib\x64
D:\NVIDIA\CUDA\bin
D:\NVIDIA\CUDA Samples\common\lib\x64
D:\NVIDIA\CUDA Samples\bin\win64
我的配置情况
系统变量
PATH
快捷键win+R,输入cmd,然后切盘到CUDA的安装路径,这里我的路径是D:\NVIDIA\CUDA
输入** nvcc -V** ,回车运行后返回版本信息,则可视为成功。
安装成功如图所示:
tensorflow 简单的测试程序
随文附上我的版本文件包【CUDA10.1+CUDnn10.1】,CUDA官网比较卡,可能需要挂梯子下载。
CUDA10.1 提取码:asdf
CUDnn 提取码:asdf
CUDA+CUDnn 提取码:asdf