Windows10搭建tensorflow教程GPU+pycharm

 

(转载请注明出处)

电脑环境:windows10 + GeForce 840M + cuda9 + cudnn7 + tensorflow1.9+python3.5

 

1、官网直接下载anaconda

2、创建一个tensorflow虚拟环境,此处命名为tensorflow-gpu(或者可以直接在命令行创建,命令如下)

conda createt -n tensorflow-gpu python==3.5

3、nVidia官网直接下载cuda9,安装,并配置环境变量

4、官网下载cudnn7,需要注册英伟达账号,注册时间比较慢,可以在网上另找资源,下载完成后,将文件解压,并按照文件夹名称分别放在cuda安装目录下,这里提供win10的cudnn7百度云链接,

链接:https://pan.baidu.com/s/15Z4wcHNid9KBNu9KTCu_EQ 
提取码:ytah

 

5、进入cmd,命令

activate tensorflow-gpu

激活环境,紧接着执行以下命令来在线安装tensorflowGPU1.9

pip install tensorflow-gpu==1.9

6、pycharm配置python interpreter

7、测试代码为自己实验代码。不展示详细,贴出控制台打印出的显卡信息,证明测试成功(显示显卡信息)

 

猜你喜欢

转载自blog.csdn.net/lqz790192593/article/details/83419823