cuda 11.6
cudnn 8.9.0
python 3.10
对应安装的gpu版本
tensorflow–gpu 2.10.0
pip install tensorflow-gpu==2.10.0 -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
具体怎么安装的我已经放在文章底部啦,改镜像源什么的也不多说~
感谢参考嘿(-v–
——————————————————————————————–分割线—————————-
>>>避坑过程
#采用清华镜像的pip安装
#pip install 命令后面带上“-i 源地址”
pip install tensorflow-gpu -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
那么能得到一大堆Error,告知你此包不存在(默认安装最新版)
不仅如此,还会善意地提醒你去安装tensorflow(即CPU版本),让你心安理得地感觉很成功,最后到跑gpu的时候发现莫名其妙的报错【巨坑】
Why?
最新版的tensorflow-gpu版本是2.12.0,国内镜像源还没上线呢!(2023.4.21记录)
2)于是你就想,既然我得不到最新版的,那么我就安装以往版本的gpu呗
但很遗憾地告诉你,这张图已经两三年没更新了,你根本找不到你电脑配置对应的最新版本
耶!!!
pip install tensorflow-gpu==对应版本号 -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
目前有的版本
2.8.0rc1, 2.8.0, 2.8.1, 2.8.2, 2.8.3, 2.8.4, 2.9.0rc0, 2.9.0rc1, 2.9.0rc2, 2.9.0, 2.9.1, 2.9.2, 2.9.3, 2.10.0rc0, 2.10.0rc1, 2.10.0rc2, 2.10.0rc3, 2.10.0, 2.10.1
虽然也不知道有什么后果,但祝你好运!
最终成功结果也展示给大家吧
import tensorflow as tf
print(tf.test.is_gpu_available())
学术小白,如有错误,还请指正哈~
#参考文章https://blog.csdn.net/sinat_39620217/article/details/115462155
#参考文章Tensorflow-gpu安装教程(详细)!!!_Hacker Belial的博客-CSDN博客
原文地址:https://blog.csdn.net/m0_62517792/article/details/130278920
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若转载,请注明出处:http://www.7code.cn/show_11213.html
如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱:suwngjj01@126.com进行投诉反馈,一经查实,立即删除!