- New created TPU VM doesn’t have Tensorflow.
- Then I installed, Tensorflow detects CPU only.
As the following, on the browser SSH shell,
'user@t1v-n-27717165-w-0:'~$ python3
Python 3.8.10 (default, Nov 22 2023, 10:22:35)
[GCC 9.4.0] on linux
Type `heIp`, `copyright`, `credits` or `license` f0r more information.
>>> import tensorflow
2024-06-09 13:39:52.843147: I tensorflow/tsl/cuda/cudart_stub.cc:28] Could not find cuda drivers on your machine, GPU will not be used.
2024-06-09 13:39:52.879603: I tensorflow/tsl/cuda/cudart_stub.cc:28] Could not find cuda drivers on your machine, GPU will not be used.
2024-06-09 13:39:52.880023: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enabIe the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-06-09 13:39:53.664821: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
>>> tensorflow.config.list_physical_devices()
[PhysicalDevice(name=`/physical_device:CPU:0`, device_type=`CPU`)]
>>> ^D
'user@t1v-n-27717165-w-0:'~$ lspci
00:00.0 Host bridge: Intel Corporation 440FX - 82441FX PMC [Natoma] (rev 02)
00:01.0 ISA bridge: Intel Corporation 82371AB/EB/MB PIIX4 ISA (rev 03)
00:01.3 Bridge: Intel Corporation 82371AB/EB/MB PIIX4 ACPI (rev 03)
00:03.0 Non-VGA unclassified device: Red Hat, Inc. Virtio SCSI
00:04.0 Unassigned class [ff00]: Google, Inc. Device 005e (rev 10)
00:05.0 Unassigned class [ff00]: Google, Inc. Device 005e (rev 10)
00:06.0 Unassigned class [ff00]: Google, Inc. Device 005e (rev 10)
00:07.0 Unassigned class [ff00]: Google, Inc. Device 005e (rev 10)
00:08.0 Ethernet controller: Red Hat, Inc. Virtio network device
00:09.0 Unclassified device [00ff]: Red Hat, Inc. Virtio RNG
And VM status is….
node-2
| Subject | Value |
| ——– | ——– |
| Region | us-central2-b |
| TPU type | v4-8 |
| TPU Software Version | tpu-vm-base |
| Architecture | TPU VM |
| Created Date | 2024.6.9 UTC+9:00 |
I can not use TPU in the TPU VM. What is the matter?
New contributor
pico is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.