为电脑配置GPU加速（win10+python3.6+tensorflow-gpu1.12.0)
192 浏览量
2021-01-20
03:17:00
上传
评论
收藏 539KB PDF 举报
为电脑配置为电脑配置GPU加速（加速（win10+python3.6+tensorflow-gpu1.12.0)
本教程基于
win10
系统，所用到的工具为
Anaconda3
，
PyCharm
你需要已经安装好以上工具，具体操作略。
首先确定你的显卡是否支持gpu加速，可在nvidia官网查看是否有你的显卡所对应的算力，如图。
为了不影响其他的工作环境，本教程创建anaconda虚拟python环境，打开Anaconda Prompt，输入指令
conda create -n your_env_name python==3.6
待安装完成后，输入指令
conda env list
conda会罗列出所有你所创建的环境，如图
输入如下指令，进入conda虚拟python环境，接下来的所有操作都在此虚拟环境下进行
activate your_env_name
如图进入python虚拟环境
安装安装
tensorflow-gpu
接下来输入如下指令安装tensorflow-gpu
pip install tensorflow-gpu
如需安装指定版本(例如1.12.0)键入
pip install tensorflow-gpu==1.12.0
安装过程会安装tensorflow-gpu及其依赖包，耗时略长，建议更换国内源，例如中科大、清华、豆瓣等
（tensorflow还有离线安装方式，本教程略）
待安装完成后并不能正常运行，会报错DLL缺少模块，此为正常现象因为还未安装cuda和cudnn。
安装安装
cuda
在tensorflow官网确定tensorflow-gpu对应的cuda和cudnn版本，如图tensorflow1.12.0对应的cuda版本为9，cudnn版本为7
评论0