本人的笔记本基本配置如下:
CPU: i7-8750H
运行内存:16G
GPU:GTX 1060
显存:6G
一、准备工作
1.查看win10中是否具有“nividia 控制面板”,一般桌面鼠标右键即可看到:
点击后如下图:
点击“系统信息”可以查看显卡信息,其中“组件”中可以查看到当前驱动对应的cuda版本:
图中显示的是nvidia驱动支持的cuda 最高版本,驱动向下兼容其它cuda 版本。
2.若电脑中没有“nvidia 控制面板”或者没有“nvidia驱动”,直接去nvidia官网下载:
3.设置好GPU权限,防止最后运行程序GPU不会被调用
4.电脑中安装好anaconda
二、安装 tensorflow-gpu==2.1
我的python版本是python3.7.9
cmd下先激活环境:conda activate xxxx
接着执行下面两句指令即可:
conda install tensorflow-gpu==2.1.0
conda install cudatoolkit=10.1 cudnn=7.6.5
三、测试
网上找个测试用例即可