文章目录
本人亲测有效,只希望可以帮到有需要的人!
本人安装版本:Tensorflow 1.13.1 CUDA 10.0 cuDNN 7.6
1 Anaconda下载安装
2 TensorFlow环境配置及安装
【注意】TensorFlow、CUDA、cuDNN版本一定要配套!!!
2.1 TensorFlow、CUDA、cuDNN版本的选择
2.2 CUDA下载安装
2.2.1 CUDA下载
选择对应的版本进行下载
2.2.2 CUDA安装
安装时需要注意一个问题,如图所示:选择自定义安装,下一步打开CUDA的折叠项,取消VS的勾选
2.3 cuDNN下载与配置
2.2.1 cuDNN下载
cuDNN 下载
选择对应的版本后,需要注册账号才能下载,按照要求注册即可。
2.2.2 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
(2)选择虚拟环境(选择anaconda3/envs/tfenv/python.exe)
(3)TensorFlow 环境配置完成!