200字范文,内容丰富有趣,生活中的好帮手!
200字范文 > win10下TensorFlow-GPU安装(GTX1660+CUDA10+CUDNN7.4)

win10下TensorFlow-GPU安装(GTX1660+CUDA10+CUDNN7.4)

时间:2022-04-10 15:22:43

相关推荐

win10下TensorFlow-GPU安装(GTX1660+CUDA10+CUDNN7.4)

首先安装anaconda

Anaconda下载地址:/download/

注意在安装时勾上第一项:

安装CUDA10.0:

CUDA下载地址:/cuda-toolkit-archive

安装CUDA前会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。

选择自定义安装,安装选项我全选了

下一步会让选择安装路径,个人建议如果C盘能接受一个多G,就直接安装在默认路径,防止出问题

安装完成之后,下一步需要配置环境变量:

计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V10_0两个环境变量,接下来,还要在系统中添加以下几个环境变量:

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0(这是默认安装位置的路径,如果自己自定义路径设置安装成功的话就用自己的路径)

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

下一步在系统变量PATH里添加路径。找到系统变量的PATH双击并添加下列几个路径:

%CUDA_LIB_PATH%

%CUDA_BIN_PATH%

%CUDA_SDK_LIB_PATH%

%CUDA_SDK_BIN_PATH%

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64 这些均为默认路径,有需要的话自行修改

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\common\lib\x64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\bin\win64

添加完之后:

添加完之后CUDA就算安装完成了。我们可以检验是否安装成功。

打开CMD,cd到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite (这是默认路径)

分别执行命令:bandwidthTest.exe

deviceQuery.exe

看到pass说明成功啦

安装cudnn7.4

CuDNN下载地址:/rdp/cudnn-download

(下载CuDNN需要注册账号,注册过程比较简单)

下载好之后,解压,分别将.h .lib 和.dll 文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0的include, lib/x64, bin 文件夹下。其实也就是对应的文件夹。

到这里就安装好了CUDA10.0及其对应的cudnn。

安装TensorFlow-GPU

打开命令行窗口,首先创建运行环境,输入指令:

conda create -n tensorflow-gpu python=3.6

上述指令新建一个名字叫“tensorflow-gpu”,python版本为3.6的运行环境

激活并进入环境,使后续指令在激活的环境中生效,输入指令:

conda activate tensorflow-gpu

升级pip到最新版,防止稍后的安装时,出现错误,输入命令:

python -m pip install --upgrade pip

安装TensorFlow,由上表可知,安装的TensorFlow的版本为1.13.1,输入命令:

pip install tensorflow-gpu==1.13.1

建议换成国内的源,这样速度会快很多:

pip install tensorflow-gpu==1.13.1 -i https://pypi.tuna./simple

验证:

在命令行中,进入python,并输入以下代码:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

输出:

b’ Hello, TensorFlow

若能正确输出则说明成功安装,但是我在输入import tensorflow as tf 后出现一连串警告:

FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'......

查找原因之后发现是numpy版本太高,所以需要将numpy版本将到1.16.0:

pip install numpy==1.16.0

到这里就已经安装完成了

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。