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 tensorflow.py 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