最近显卡降价,入手了一块RTX3060TI战斧显卡,美滋滋,顺便升级一下深度学习的单机配置
特别说明的是30系列显卡之后必须用cuda11.0以上,因为之后的架构发生了改变,否则30系显卡无法调用
这里我用的环境是
可是怎么事先知道那个版本的windows系统,显卡型号,tensorflow,cuda,cudnn是相互配合的呢?
先看显卡,看看显卡支持到哪里?
找到nvidia控制面板,看一下支持的cuda版本信息,要小于显卡所标识的最大版本号
cmd输入nvidia-smi,查询驱动器版本号,
那么这里选择cuda11.0是可行的,然后找官方给的配置单,看一下相互对应的tensorflow等等
去英伟达官网,他们给的有版本配置单,按着来就行了
https://tensorflow.google.cn/install/source?hl=en#gpu
那么知道为什么用这些文件了,就进入文件安装使用吧
cuda安装需要vs环境,适配的组件在vs2019中都具备了,因此要先安装vs2019
需要注意的是在安装时勾选以下组件,否则还是没有cuda运行环境
从网上下载个vs2019安装器,然后勾选如下即可,等待安装完成
cuda第一开始显示的文件路径框是自解压,选择一个空白文件夹即可,安装完毕后其会将解压的文件自动删除
然后没什么可说的,全部勾选,下一步下一步
如果有人报错了,看一下安装vs2019时的两个运行环境勾选没有,或者看看你的显卡的支不支持这个cuda文件?
其实就是解压的文件夹,将cudnn中的三个文件复制放入cuda的安装目录如下
cudnn解压后的文件夹,全选复制一下
粘贴到cuda安装目录
使用清华镜像更快一些
不需要清华镜像就把后面网址删掉
pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #版本可任意修改
GPU是否识别
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
下一篇:进程地址空间(虚拟地址空间)