安装Tensorflow-gpu时遇到如下问题:
1.(base)环境下,pip install tensorflow-gpu查询不到我想要安装的2.8.0版本
通过搜索发现,tensorflow已经将cpu和gpu集成,无需分别安装,直接输入如下:
“pip install tensorflow==2.8.0”
2.(base)环境下,报错:
“ERROR: Could not find a version that satisfies the requirement tensorflow==2.8.0 (from versions: 2.12.0rc0, 2.12.0rc1, 2.12.0, 2.12.1, 2.13.0rc0, 2.13.0rc1, 2.13.0rc2, 2.13.0, 2.13.1, 2.14.0rc0, 2.14.0rc1, 2.14.0, 2.14.1, 2.15.0rc0, 2.15.0rc1, 2.15.0)
ERROR: No matching distribution found for tensorflow==2.8.0”
怀疑是旧版本不再提供,因此尝试安装2.12.0版本
pip install tensorflow==2.12.0
安装成功。
3.在pycharm中运行tensorflow,验证是否能成功调用gpu,报错:
Num GPUs Available: 0
解决方案如下:
1.在pycharm中创建新的解释器(tensorflow)
在setting→Python Interpreter→Add Interpreter→Add Local Interpreter→Environment选择Existing→Location定位到Anaconda文件夹下的envs\tensorflow\python.exe
确认自己的pycharm运行环境是新创建的(tensorflow)
然而,此时运行程序,报错:
“ModuleNotFoundError: No module named 'tensorflow'”
2.确认(base)环境和(tensorflow)环境中是否有tensorflow包
在Anaconda Prompt中依次确认:
(base)环境
conda list
(tensorflow)环境
conda activate tensorflow
conda list
发现,(base)环境下存在安装的tensorflow 2.12.0
而,(tensorflow)环境下没有找到tensorflow包
因此,需要在(tensorflow)环境下安装,尝试安装我需要的2.8.0
pip install tensorflow==2.8.0
可以连接到!!!此环境下,能查找到2.8.0及以上版本。(而在(base)环境下仅查找到2.12.0及以上版本)
3.回到pycharm中确认是否能查找到gpu
import tensorflow as tf
tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()
print('tensorflow version:', tensorflow_version, '\tGPU available:', gpu_available)
成功!!!