TensorFlow环境配置【Win 10+GPU】


本人亲测有效,只希望可以帮到有需要的人!
本人安装版本:Tensorflow 1.13.1 CUDA 10.0 cuDNN 7.6

1 Anaconda下载安装

Anaconda下载安装参考

2 TensorFlow环境配置及安装

【注意】TensorFlow、CUDA、cuDNN版本一定要配套!!!

2.1 TensorFlow、CUDA、cuDNN版本的选择

版本对应关系
CPU各版本对应关系

2.2 CUDA下载安装

2.2.1 CUDA下载

CUDA 下载

选择对应的版本进行下载CUDA下载

2.2.2 CUDA安装

安装时需要注意一个问题,如图所示:选择自定义安装,下一步打开CUDA的折叠项,取消VS的勾选
CUDA安装
CUDA安装

2.3 cuDNN下载与配置

2.2.1 cuDNN下载

cuDNN 下载
选择对应的版本后,需要注册账号才能下载,按照要求注册即可。

2.2.2 cuDNN配置

cuDNN解压后如图
cuDNN
将cuDNN包含的文件拷贝至CUDA文件夹下,我的地址是

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

2.4 TensorFlow-GPU安装

(1) 进入创建好的虚拟环境

activate tfenv

(2)安装tensorflow-gpu

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

(3)检查

import tensorflow as tf

3 PyCharm导入虚拟环境

(1)Settings->Project
Setting
(2)选择虚拟环境(选择anaconda3/envs/tfenv/python.exe)
环境选择
(3)TensorFlow 环境配置完成!

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值