Wednesday, September 27, 2023

First python application using my GPU

 I simply wanted to make sure that some Python code could use my GPU and it took a little more effort than I taught.

The code I found simply makes sure you have some GPU available and fills in a matrix of 1000x1000. Very simple.

First rookie mistake was to call my file which cause a circular reference that took me too long to figure out.

I also needed to install a few missing components on my fresh Ubuntu 22.04 installation.

pip install tensorflow

sudo apt install nvidia-cuda-toolkit

pip install tensorrt

Once all these components were present, the application ran without errors.

No comments:

Post a Comment