Tensorflow学习:win10安装tensorflow-gpu
1、安装CUDA
1)确认自己的笔记本带有NVIDIA的显卡。电脑->属性->设备管理器->显示适配器
2)下载安装CUDA 。https://developer.nvidia.com/cuda-downloads
下载的安装包如下:
安装过程 和安装其他应用一样,直接下一步下一步,中间可能会有闪屏的现象。
3)安装好之后把CUDA安装目录下的bin和lib\x64添加到 Path环境变量中。
bin路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
x64路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64
配置环境变量:
安装测试:cmd
C:\Users\zhzsdiligence>cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Thu_Jun_11_22:26:48_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.194
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28540450_0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin>
2、安装cuDNN
1)cuDNN下载 。https://developer.nvidia.com/rdp/cudnn-archive
这个下载需要先注册才能下载,网盘下载好的:链接: https://pan.baidu.com/s/1L8m7PDzzKJb457q7FxQzrw 提取码: 1irj
2)解压压缩包,把压缩包中bin,include,lib中的文件 分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0目录下对应目录中 。
3)把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\CUPTI\lib64下的cupti64_2020.1.0.dll拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin。
3、安装tensorflow-gpu
卸载之前的TensorFlow(cpu)
安装TensorFlow-gup
cmd:
1.pip uninstall tensorflow
2.pip install tensorflow-gpu