我想将tf更新到2.7.0,通过命令更新tf后:
检查是否升级成功:
结果是没有找到可用的 GPU:
这是为什么呢? 以及如何正确升级tensorflow的版本?
首先,通过
更新Kaggle内核中的Tensorflow版本!pip install -U tensorflow==2.3.1
然后,更新GPU集群中的GPU版本:
print("update GPU server TensorFlow version...")
!pip install cloud-gpu-client
import tensorflow as tf
from cloud_gpu_client import Client
print(tf.__version__)
Client().configure_tpu_version(tf.__version__, restart_type='ifNeeded')
按CTL+SHIFT+P并选择重新启动内核
Kaggle 本身有指南解释这些库在其内核上的用法。
但是,现在认为这已经足够了。