摘要:博主前阵子不小心升级了tf到2.13.0,但是发现GPU无法使用了,官网查看发现tf-gpu最多支持到2.10版本,故退回版本。
1.通过此代码可以查看GPU是否可用,返回1-n任意一个数字即为可用,返回0不可以。
import tensorflow as tf
print("Num of GPUs available: ", len(tf.test.gpu_device_name()))
2.查看自己的tf版本,cuda版本,cudann版本是否一致,注意tf版本最高到2.10.
3.关于回退tf版本:第一行代码删除tf,第二行下载2.10版本tf
pip uninstall tensorflow
pip install tensorflow==2.10
楼主操作完两行代码后发现successfully installed,但是jupyternotebook无法使用tf,且版本显示‘2.13.0’.在anaconda prompt输入如下代码,仍然显示‘2.13.0’,安装不完全成功。但由于上述成功安装tf,故问题不在安装上。
python
import tensorflow as tf
tf.__version__
4.发现问题:
考虑路径问题,输入下面代码查看路径,楼主之前把anaconda下载到D盘中过,所以路径显示两个,一个C盘一个D盘
tf.__path__
分别去下面几个路径看看,如果上述正确pip install过tf,D盘是没有问题的(通过观察修改日期即可查明是否有过改动),C盘的tf并没有删除,还是原始的tf2.13.0。
5.解决问题:
打开anaconda prompt,输入下面代码(用镜像直接下载tf2.10)
pip install tensorflow==2.10 -i https://pypi.tuna.tsinghua.edu.cn/simple
下载完成后去上述D盘的位置,我这里是D:\\anaconda3\\lib\\site-packages
找到两个文件,tensorflow和tensorflow2.10.0.dist-info。复制粘贴到上述C盘的位置,替换C盘中两个文件名一样的文件,替换完成后即可使用。