NVTX missing using SE3nv.yml -- Pytorch 2.0 solution

Device

OS: CentOS Linux 7
GPU: gtx 1080

Issue

Hi! I get the following error running any of the examples scripts

RuntimeError: NVTX functions not installed. Are you sure you have a CUDA build?

When using the current SE3nv.yml I get the following versions

pytorch                   1.9.1           cpu_py39hc5866cc_3    conda-forge
torchaudio                0.9.1           py39                  pytorch
torchvision               0.14.1          cpu_py39h39206e8_1    conda-forge

Solution

I did a clean install running pip3 install --force-reinstall torch torchvision torchaudio

torch                     2.0.0                    pypi_0    pypi
torchaudio                2.0.1                    pypi_0    pypi
torchvision               0.15.1                   pypi_0    pypi

That seems to run every example without an issue. I've come into issues before with conda installs for pytorch when not using the most recent version. Is there a known issue from keeping RFdiffusion from moving to pytorch 2.0?

Start by inspecting SE3nv.yml and the example scripts, then compare the CPU-only PyTorch 1.9.1 packages installed from conda with the PyTorch 2.0.0 packages that run successfully. Reproduce the NVTX error and determine what environment change is needed; done when the examples run without the missing-NVTX failure.