Skip to content

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

Answered by roccomoretti
zarbuk asked this question in Q&A
Discussion options

You must be logged in to vote

Generally this means that you have the CPU-only build of PyTorch installed in your Conda environment.

First off, do you have an NVIDIA CPU with CUDA support? Usually you can run nvidia-smi to get a report, and see the maximum CUDA version your hardware supports. For the provided environment files, this should be 11.1 or greater. If your max supported CUDA version is less than that, or you don't have an CUDA-capable NVIDIA GPU, then you'll have to adjust your conda installation for it.

If you do have at least CUDA 11.1 support, then try a conda list | grep torch with the RFDiffusion environment activated. You should see "pytorch" as one of the lines. You should be able to also see the deta…

Replies: 3 comments 6 replies

Comment options

You must be logged in to vote
5 replies
@zarbuk
Comment options

@roccomoretti
Comment options

@zarbuk
Comment options

@roccomoretti
Comment options

@zarbuk
Comment options

Answer selected by zarbuk
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
1 reply
@roccomoretti
Comment options

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
3 participants