最近因為學習需要，所以在看別人的代碼，但是發現有些教科書或是很久以前的代碼都是在TF1.x的環境下撰寫的，而很多顯卡例如Genforce系列在TF1.x中是無法派上用場的，因此需要安裝TF2.x-gpu版本來運行代碼，但是中間也遇到了很多問題：
系統：win11
cuda 11.2
tensorflow-gpu 2.6
下面來講講我的解決程序，首先在安裝完對應版本的cuda和cudnn后，安裝TF2.x-gpu后即可，但是會發現有諸多不兼容的地方，所以參考別人在Ubuntu環境下的解決方案給出自己在Win11環境下的流程，
首先，使用
import tensorflow as tf
import tensorflow.compat.v1 as tf
//使用上述陳述句來改用TF1.x
之后需要使用下述陳述句來禁用TF2.x新的特性
tf.disable_v2_behavior()之后在運行時報錯，“No module named tf_slim",而tf_slim是tensorflow1.x版本中的瘦身，因此在tf1.12以上的版本中似乎沒有了，因此直接使用
pip install tf_slim后成功解決！
成功解決后，報錯處會顯示No module named compat,但是實際上不影響運行，以上為在TF2.x環境下運行TF1.x代碼的流程，
參考文章：
(17條訊息) TensorFlow2運行TF1程式并解決GPU無法使用的問題_JeremyZhao1998的博客-CSD博客???????s
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/430259.html
標籤：AI