查阅官网后可知,该命令已经被删除掉了。

tensorflow-gpu · PyPI

【解决办法】

我直接安装了其他版本的包

pip install tensorflow-gpu==2.10.0

测试

import tensorflow as tf

print("tf.__version__: ", tf.__version__)

# print("tf.test.is_gpu_available(): ", tf.test.is_gpu_available())

print("tf.config.list_physical_devices('GPU'): ", tf.config.list_physical_devices('GPU'))

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐