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!