search

NVIDIA / nvtx-plugins

Python bindings for NVTX
https://docs.nvidia.com/deeplearning/frameworks/nvtx-plugins/user-guide/docs/en/stable/
Apache License 2.0
66 stars 15 forks source link

nvtx-plugins installation failed with TensorFlow 2.6 #26

Open Jianbing-D opened 3 years ago

Jianbing-D commented 3 years ago

In TF 2.6, install nvtx-plugins will fail.

Produce steps:

  1. use this docker image, where TF 2.6 is installed. nvcr.io/nvidia/tensorflow:21.10-tf2-py3
  2. pip install nvtx-plugins