一、自己的环境
操作系统:win10 GPU:GTX1050+CUDA9.0+cuDNN IDE:Pycharm 框架:tensorflow-gpu 解释器:Python3.6(强烈建议anaconda3,又方便又强大,下载太慢可以找镜像)
二、安装顺序
1、首先安装CUDA9.0。下载地址。
选择9.0版本是因为网上说,tensorflow的windows-gpu还不支持CUDA的新版本。再加上同学安装的也是9.0,用的没有问题,所以就选择了这个版本。
在下载的时候选择适合自己的版本。
我选择的如上图,刚开始第四行选择的是右边的那个,但是下载下来安装不了,百度查是电脑缺少什么压缩软件,所以建议选左边的那个。
然后只需要选择下图的base installer即可。
打开下载好的CUDA9.0开始安装,刚开始的安装程序存放位置,自己随意就好。
win10搭建tensorflow-gpu环境
最新推荐文章于 2024-10-02 11:45:00 发布
本文档详述了在Windows 10系统中,如何配置CUDA9.0和cudnn以安装TensorFlow-GPU的步骤。包括CUDA的选择与安装、环境变量配置,以及cudnn的下载与放置。最后,通过测试验证CUDA和TensorFlow-GPU的安装成功。
摘要由CSDN通过智能技术生成