RuntimeError: Cannot find ptxas

I’m using the NVIDIA Docker container and the latest pytorch nightly:

nvcr.io/nvidia/cuda:11.8.0-devel-ubuntu22.04

pip3 install --pre torch torchvision torchaudio --force-reinstall --index-url https://download.pytorch.org/whl/nightly/cu118

torch.__version__ == 2.0.0.dev20230218+cu118

I can no longer compile after updating to the latest pytorch nightly.

This is the error:

RuntimeError: Cannot find ptxas

I saw this on twitter:

so presumably this is just a config/path error? How should I go about fixing this?

Thanks!

Our latest rebuild as of yesterday should have fixed it, but let me double check it.

EDIT: Something still seems to be broken in the latest build. We are tracking the issue here.

1 Like