tensorflow-gpu的安装教学

大佬的CSDN链接

https://blog.csdn.net/qq_43215538/article/details/123852028?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165231684916782350913961%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=165231684916782350913961&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-123852028-null-null.142^v9^control,157^v4^control&utm_term=tensorflow-gpu&spm=1018.2226.3001.4187

1. 查看tensorflow对应版本所需要的cuda版本和cuDNN版本，在下面链接网站的底部可以查看。

　　https://tensorflow.google.cn/install/source_windows

2. 安装cuda

　　首先查看本机GPU对应的cuda版本，如下图所示，本机cuda版本为11.5，后面选择的cuda版本不要超过这里的版本就好。这是非常重要的，不然电脑硬件将不支持，无法运行。

　　历史版本链接：https://developer.nvidia.com/cuda-toolkit-archive

3. 第二步：安装cuDNN

　　cuDNN历史版本链接

　　https://developer.nvidia.com/rdp/cudnn-archive

需要将cuDNN的文件复制粘贴到cuda文件夹

4. 安装tensorflow

　　pip install tensorflow-gpu==1.15.0 --default-timeout=100 -i https://pypi.tuna.tsinghua.edu.cn/simple    版本可以根据自己的需求进行更改。

 

 

 

在跑代码中还会遇到各种各样的问题，慢慢去探索吧。

1. 在代码里增加显卡内存自动分配

　　import os

　　os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'

2. tensorflow与keras的版本对应关系

　　https://blog.csdn.net/weixin_44052055/article/details/122256557

3. AttributeError: ‘str’ object has no attribute 'decode

　　解决办法：print (‘xxx’.encode(‘utf-8’). decode(‘utf-8’) ) ,即先encode，再decode