安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本


windows上安装gpu版本tensorflow2.0环境时,遇到了一些问题,这里记录下来供大家参考。 在安装gpu版 t ensorf low2.0 之前,需要先安装相应版本的cuda和cudnn,各版本对照表如下:

安卓怎么看GPU占用 安卓查看gpu驱动版本_安卓怎么看GPU占用_02


1、更新显卡驱动

不同版本的cuda对显卡驱动版本有要求,所以需要先安装对应版本的显卡驱动。具体对照如下:

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_03

查看本机显卡驱动版本,方法有2种,一种方法是按下图方法查看:

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_04

另一种是通过命令行的方式查看:

nvidia-smi

用命令行方式之前,需要先配置环境变量PATH,一般PATH为:

C:\Program Files\NVIDIA Corporation\NVSMI

PATH配完之后,重启命令行就可执行查看:

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_05

安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本_06

如果驱动不符合,可以在官网:

https://www.geforce.cn/drivers

下载安装,推荐自动更新方式,安装完成后,最好重启下电脑。

安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本_07


2、安装cuda

在官网下载对应版本的cuda然后进行安装,安装完成后需要配置2个PATH环境变量 (具体见下图) 。

安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本_08

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_09


配置好环境变量后,执行命令检查是否安装成功:

nvcc -V

安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本_10


3、替换cudnn文件

cudnn是神经网络的加速包,可以去官网下载对应版本的cudnn压缩包。解压cudnn文件后,需要用加压出来的3个 文件夹中的文件(非整个文件夹)

把cuda安装路径的相应文件夹下的同名文件替换掉。 (注意lib文件夹还有个2级文件夹,只需替换里面的1个文件)

安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本_11

安卓怎么看GPU占用 安卓查看gpu驱动版本_cudnn版本_12


4、安装tensorflow-gpu==2.0

安装tensorflow包的时候,用管理员权限启动命令行,不然可能会遇到权限不够的情况。pip install后面最好加上国内的源,下载速度会快很多:

安卓怎么看GPU占用 安卓查看gpu驱动版本_安卓怎么看GPU占用_13

pip install tensorflow-gpu==2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple


如果本机上以前安装了版本不符合的wrapt时,在卸载升级的时候可能会遇到编码问题报错,这时候可以根据报错提示,把此路径: C:\ProgramData\Anaconda3\Lib\site-packages\pip\compat\__init__.py 第75行的s.decode(' utf_8 '),先改成s.decode(' gbk '),等安装成功后,再改回去。

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_14

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_15

等tensorflow安装成功后,检查一下tf版本是否正确:

import tensorflow as tftf.__version__

安卓怎么看GPU占用 安卓查看gpu驱动版本_diskgeniusv4.4.0_16

没有报错,安装成功!


参考资料:


https://tensorflow.google.cn/install/source#linux

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

https://developer.nvidia.com/cuda-10.0-download-archive