Windows10系统tensorflow-gpu安装

2021-06-30 04:05

阅读：599

标签：archive 根据 VID dnn src href pat 技术分享 har

安装前请确保自己的显卡支持gpu加速，支持加速的gpu型号可在下面的链接中查询。

https://www.geforce.com/hardware/technology/cuda/supported-gpus

一、安装anaconda

下载安装cuda，https://developer.nvidia.com/cuda-downloads
下载cudnn，https://developer.nvidia.com/cudnn ，（下载cudnn需要注册账号，请根据cuda版本选择相对应cudnn）
解压cudnn，将解压出来的三个文件夹下面的文件移动到对应的cuda文件夹下（cuda默认安装路径为 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0）　　　　
将cuda的lib和include文件夹添加到环境变量

　　打开命令行，执行下面命令。

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

Windows10系统tensorflow-gpu安装

标签：archive 根据 VID dnn src href pat 技术分享 har

原文地址：https://www.cnblogs.com/clearlove7777777/p/9996918.html

文章来自：搜素材网的编程语言模块，转载请注明文章出处。
文章标题：Windows10系统tensorflow-gpu安装
文章链接：http://soscw.com/index.php/essay/99675.html

亲，登录后才可以留言！