Skip to content

Tensorflow CUDA DevContainer Configuration for Supporting NVIDIA GPU

Notifications You must be signed in to change notification settings

Hector-Work/tensorflow-cuda-gpu-devcontainer

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

41 Commits
 
 
 
 

Repository files navigation

TensorFlow CUDA GPU Devcontainer

To remove the NUMA node error, run:

for a in /sys/bus/pci/devices/*; do echo 0 | sudo tee -a $a/numa_node; done

To test that the TensorFlow installation is detecting a GPU, run:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

About

Tensorflow CUDA DevContainer Configuration for Supporting NVIDIA GPU

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Shell 100.0%