Installing GPU-Aware Software

You likely know about certain software packages which are frequently used with GPUs. You might NOT know that, by default, some of those packages default to only using the CPU, even if a GPU is present.

Please refer to the instructions below to install these packages such that they’ll take advantage of the GPU.

Installing TensorFlow

pip3 install "tensorflow[and-cuda]"

More info: https://www.tensorflow.org/install/pip

Installing PyTorch

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130

More info: https://pytorch.org/get-started/locally/