I would like to enable NVIDIA T600 Laptop GPU for tensorflow and python code in Visual studio code

AI & Data Science Deep Learning (Training & Inference) TensorRT
,
1

Description

I would like to enable NVIDIA T600 Laptop GPU for tensorflow and python code in Visual studio code.

Environment

TensorRT Version: N/A.
GPU Type:
1.GPU 0 Intel(R) UHD Graphics
2.GPU 1 NVIDIA T600 Laptop GPU.
Nvidia Driver Version:
1.GPU 0 Intel(R) UHD Graphics : 31.0.101.4091
2.GPU 1 NVIDIA T600 Laptop GPU. : 32.0.15.7216
CUDA Version:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2025 NVIDIA Corporation
Built on Wed_Jan_15_19:38:46_Pacific_Standard_Time_2025
Cuda compilation tools, release 12.8, V12.8.61
Build cuda_12.8.r12.8/compiler.35404655_0
CUDNN Version: Download cuDNN v8.9.7 (December 5th, 2023), for CUDA 12.x

Operating System + Version: Windows 10 Enterprise 64-bit (10.0, Build 19045)
Python Version (if applicable): Python 3.11.1
TensorFlow Version (if applicable): 2.18.0
PyTorch Version (if applicable): N/A
Baremetal or Container (if container which image + tag): N/A

Relevant Files

Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.) : Below code.

import tensorflow as tf
print("Num GPUs Available: “, len(tf.config.list_physical_devices(‘GPU’)))
print(tf.config.list_physical_devices(‘GPU’))
build_info = tf.sysconfig.get_build_info()
print(“CUDA Version:”, build_info.get(“cuda_version”))
print(“cuDNN Version:”, build_info.get(“cudnn_version”))
print(f"tensorflow version: {tf.version}”)

import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000]))) # If tensor existing meaning CPU is work!

import tensorflow as tf; print(tf.config.list_physical_devices(‘GPU’)) # If tensor existing meaning GPU is work! in this cause GPU did not work.

Results :
Num GPUs Available: 0

CUDA Version: None
cuDNN Version: None
tensorflow version: 2.18.0
tf.Tensor(2523.2764, shape=(), dtype=float32)

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro : Run above python command .ipynb.
  • Exact steps/commands to run your repro : Run above python command .ipynb…
  • Full traceback of errors encountered.
4

Hi @jongruk.ritu ,

This forum talks about issues related to TRT, and I am afraid I might not be able to assist you on this.
Would recommend you to raise it under HArdware category.
Thanks

5

Hi @AakankshaS and NVIDIA team,

Thank you for your response. I understand that this forum focuses on TensorRT, but I would really appreciate any guidance on enabling my NVIDIA T600 Laptop GPU for TensorFlow in Visual Studio Code.

Could you kindly suggest the appropriate forum or resources where I can get support for this issue? Any recommendations would be greatly appreciated.

Thanks in advance for your help!

Sincerely,
Jongruk