A Simple & Flexible Cross Framework Operators Toolkit
-
Updated
Sep 27, 2020 - Python
A Simple & Flexible Cross Framework Operators Toolkit
This script checks the availability of CUDA-enabled GPUs and prints detailed GPU information for both PyTorch and TensorFlow frameworks. It's a handy utility for ensuring that your deep learning environment is correctly configured to utilize GPU acceleration.
Add a description, image, and links to the cross-framework topic page so that developers can more easily learn about it.
To associate your repository with the cross-framework topic, visit your repo's landing page and select "manage topics."