個人電腦相關(guān)配置版本信息
(超級超級新的版本,以至于適配方面花了很長時間來搞)
cuda? 11.6
cudnn ?8.9.0
python ?3.10
對應(yīng)安裝的gpu版本
tensorflow-gpu ?2.10.0
對應(yīng)代碼
pip install tensorflow-gpu==2.10.0 -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
具體怎么安裝的我已經(jīng)放在文章底部啦,改鏡像源什么的也不多說~
感謝參考嘿(-v-
--------------------------------------------------------------------------------------------分割線----------------------------
>>>避坑過程
1)如果按照這段代碼安裝
#采用清華鏡像的pip安裝
#pip install 命令后面帶上“-i 源地址”
pip install tensorflow-gpu -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
那么能得到一大堆Error,告知你此包不存在(默認安裝最新版)
?不僅如此,還會善意地提醒你去安裝tensorflow(即CPU版本),讓你心安理得地感覺很成功,最后到跑gpu的時候發(fā)現(xiàn)莫名其妙的報錯【巨坑】
Why?
最新版的tensorflow-gpu版本是2.12.0,國內(nèi)鏡像源還沒上線呢!(2023.4.21記錄)
2)于是你就想,既然我得不到最新版的,那么我就安裝以往版本的gpu唄
然后你大概率能在網(wǎng)上找到一張比對圖
?但很遺憾地告訴你,這張圖已經(jīng)兩三年沒更新了,你根本找不到你電腦配置對應(yīng)的最新版本
耶?。?!
3)接下來是瞎蒙版本的過程,中獎概率取決于個人手感~
pip install tensorflow-gpu==對應(yīng)版本號 -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
?目前有的版本
versions:
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
如果蒙錯了,大致會報錯(燒了我大概3GB流量
雖然也不知道有什么后果,但祝你好運!
最終成功結(jié)果也展示給大家吧
在ipython交互控制臺輸入以下代碼
import tensorflow as tf
print(tf.test.is_gpu_available())
得到True的輸出便表示你成功了!
學術(shù)小白,如有錯誤,還請指正哈~
#參考文章?????https://blog.csdn.net/sinat_39620217/article/details/115462155文章來源:http://www.zghlxwxcb.cn/news/detail-480392.html
#參考文章Tensorflow-gpu安裝教程(詳細)?。?!_Hacker Belial的博客-CSDN博客文章來源地址http://www.zghlxwxcb.cn/news/detail-480392.html
到了這里,關(guān)于解決CUDA 11.6版本對應(yīng)的tensorflow-gpu版本問題的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!