tensorflow是第二代的人工智能学习系统,被多次的用于深度学习领域,但是tensorflow安装却令人很是头疼,(当然是指gpu版本),cpu版本的过于简单,在这里就不叙述。
下面是gpu安装的具体步骤:
1:检查自己电脑上gpu是否支持cudn,当然判断是否支持cudn也很简单粗暴,就是是不是英伟达的显卡就行了,当然你也可以去英伟达网站里面查找。
具体网址:https://developer.nvidia.com/cuda-gpus
2:下载并安装cudn:进入网站:https://developer.nvidia.com/cuda-downloads 点击windows,之后选择电脑系统版本
选择版本,及安装类型(install type) 这里建议选择local,实测network有可能装不上(网速问题)。
下载完成后,一路确认默认安装到底,然后cudn就安装好了
3:下载cudnn
首先进入官网,这里要先进行注册,老老实实注册完进行安装
网址:https://developer.nvidia.com/rdp/form/cudnn-download-survey
安装后进行解压
在C盘根目录下新建tools
tensorflow-gpu安装
最新推荐文章于 2024-07-25 01:20:01 发布
本文详细介绍了tensorflow-gpu的安装步骤,包括检查GPU兼容性、下载安装CUDA和cuDNN、设置环境变量、安装Anaconda、创建tensorflow-gpu工作环境以及测试安装成功的方法。
摘要由CSDN通过智能技术生成