TensorFlow-gpu-2.5已发布【附下载地址和安装教程】

TensorFlow-gpu 2.5已发布,但官方网页还没有通知。
该版本已由google开发人员Mihai Maruseac发布在pypi网站
在这里插入图片描述

不过已经上线,查看地址链接:
地址一:
https://libraries.io/pypi/tf-nightly-gpu

地址二:
https://pypi.org/project/tf-nightly-cpu/

如果不想直接下载那就选择阿里云镜像源安装即可,安装之前请检查自己的cuda,nvidia,cudnn版本匹配情况,这里就不做介绍:可以参考我的博客文章:

(ubuntu版本)快速下载NVIDIA最新显卡驱动RTX3080ti、GTX1650ti、CUDA11.2,CUDNN的地址

赶快收藏:快速安装PyTorch和TensorFlow(gpu+cpu+1.7.1+2.2.0)命令

【长篇博文】Docker学习笔记(四):镜像的分层、打包、双向绑定、导入导出、Dockerfile的构建、部署与发布

【长篇博文】Docker学习笔记与深度学习环境的搭建和部署(二)

安装命令:

pip install tf-nightly-gpu==2.5.0 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

测试是否安装成功:
安装完成后,进入Python环境,验证是否安装成功:
代码如下:

import tensorflow as tf
print(tf.test.is_gpu_available())

如果就是True说明安装成功,gpu已经可以使用。

参考文献:
https://pip.pypa.io/en/stable/reference/pip_install/#hash-checking-mode
https://blog.csdn.net/scp_6453/article/details/110715974

猜你喜欢

转载自blog.csdn.net/weixin_41194129/article/details/114238772