CPU跑到百分百有点糊而且慢,遂下定决心安装TensorFlow-gpu。
话说TensorFlow2.0beta0(支持CPU)和beta1(支持GPU)也出了,但因为教程啥的都是1.X,就还没有研究(说实话好像用pytorch的更多啊……但再学个框架真的累)。那咱回到正题,来说咋配置吧。
首先看看咱们的显卡是什么型号。显卡和CUDA的版本对应即可。桌面点击鼠标右键,选择"NVIDIA"控制面板,就能在醒目位置看到自己的显卡版本了。有的博客会说点左下角系统信息,然后打开组件,看到3D设置中的NVCUDA.DLL:
但这是你已安装的CUDA啊!这不是你应该安装的CUDA版本。
真的要看自己应该装什么版本的CUDA,应该去英伟达的官网看CUDA对应版本的。
我的显卡是GTX1060,可以支持10.0(实际上我可以用10.1……但配windows系统的时候还不知道嘛)。
下面是cuda和cudnn的下载链接,包含8.0,9.0,10.0:
链接:https://pan.baidu.com/s/19E9D0mt7ZfubcAYBxi_LAw
提取码:xwxh
下载之后解压(cuda10.0是个文件夹,双击setup.exe安装);之后会搜索硬件,搜索不出来也可以强装(但最好不要吧……如果对应应该都是能检索出来的);安装cuda的时候可以不安装到C盘;
安装完cuda之后解压cudnn,把三个文件夹中的文件(下图中的三个)复制到cuda安装地址的同名文件夹中即可。
按理说之后就不需要什么操作了,但我在pycharm中运行文件还是编译错误。我改了一通环境变量也木大。搞到凌晨两点也没成,但睡醒之后一重启就OK了,可能就是要重启一下,安装完之后也能环境变量是自动设置好的。
8.20补充:有一位同学安装的时候安装到默认路径了,可能会有点难找,那附一张默认路径的图吧。