如何在kaggle上正确更新tensorflow?

evrscar2  于 2023-10-23  发布在  其他
关注(0)|答案(1)|浏览(192)

我想更新tf到2.7.0,通过命令更新tf后:

我检查升级是否成功:

没有找到可用的GPU:

为什么会这样?如何正确升级tensorflow的版本?

jum4pzuy

jum4pzuy1#

首先,更新Kaggle内核中的Tensorflow版本,

  1. !pip install -U tensorflow==2.3.1

然后,更新GPU集群中的GPU版本:

  1. print("update GPU server TensorFlow version...")
  2. !pip install cloud-gpu-client
  3. import tensorflow as tf
  4. from cloud_gpu_client import Client
  5. print(tf.__version__)
  6. Client().configure_tpu_version(tf.__version__, restart_type='ifNeeded')

按CTL+ CTRL +P并选择重新启动内核
Kaggle本身有指南解释这些库在其内核上的用法。
不过,目前这已经足够了。

展开查看全部

相关问题