将numpy升级至最新版本。
示例代码:
import numpy as np
import tensorflow as tf
# 检查numpy版本
print(np.__version__)
# 若numpy版本低于1.16,则升级至最新版本
if int(np.__version__.split('.')[1]) < 16:
!pip install numpy==1.19.5
# 测试TensorFlow是否可用
tf.test.is_built_with_cuda()