Win10系统Tensorflow(GPU版)安装

1.Python安装

2.CUDA和CuDNN

检查自己的显卡支持哪个版本的cuda:
控制面板—>搜索NVIDIA—(系统信息)图片1—>组件里找到支持的cuda版本(图片2)
图片1在入图片描述图片2
在这里插入图片描述
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
找到对应的cudu版本下载如图3
图片3
在这里插入图片描述
CuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-download
找到对应的CuDNN下载,并

3.TensorFlow

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

猜你喜欢

转载自blog.csdn.net/qq122716072/article/details/89792030