tensorflow-gpu 下载和安装

川长思鸟来 2022-04-22 15:04 281阅读 0赞

如果你想要GPU 进行训练的话，那你必须要有三个东西：  
1.tensorflow-gpu  
超时的话pip install tensorflow-gpu==1.13.1 -i https://pypi.douban.com/simple  
2.cuda  
CUDA其他版本下载地址是https://developer.nvidia.com/cuda-toolkit-archive  
3.cudnn  
下载地址是：https://developer.nvidia.com/rdp/cudnn-archive，在下载CUDNN7.0之前，会提示你注册一个账号，输入邮箱等相关注册信息注册一个账号  
注意：tensorflow-gpu，cuda和cudnn这两个版本号有着相互依赖的关系  
CUDA8.0 不支持tensorflow 1.5及以上，所以建议安装CUDA9.0  
安装的话就是在官网下载安装  
下载过之后：  
![在这里插入图片描述][20190114095252163.png]  
需要安装这两个 ，然后将这两个分别解压安装，解压安装之后，重点来了：  
**！！！两点必须要做！！！**：

1.  必须要将cudnn里面的bin，include，lib三个文件里面的具体的文件分别放入到安装之后的cuda里面，记住要一一对应，cudnn bin里面的文件要对应放入cuda里面的bin文件中等  
    ![20190114100823498.png][]  
    cudnn里面的三个文件
2.  环境变量的配置：  
    A.![20190114100423903.png][]  
    上面的是在安装好之后就有的，  
    ![20190114100549583.png][]  
    ![在这里插入图片描述][20190114100605238.png]  
    这个上面是你要手动添加的 注意路径问题

通过之上的安装之后，就可以安装GPU了，亲测有效![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyMDA0Mjg5_size_16_color_FFFFFF_t_70]

[20190114095252163.png]: /images/20220327/664749e6b21044139b3c7aa3979e2b35.png
[20190114100823498.png]: /images/20220327/62b56f30261248c48bfb65d739f972b5.png
[20190114100423903.png]: /images/20220327/c23d9cc47fb94ffd9f0caa4413a7f0f3.png
[20190114100549583.png]: /images/20220327/c77974568a0b4b53845e4a581762e9dd.png
[20190114100605238.png]: /images/20220327/8842fa56976b4d65b260bbcf315d38ab.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyMDA0Mjg5_size_16_color_FFFFFF_t_70]: /images/20220327/f598dc31216843d89686bfb8383f73c5.png