AOTI tutorial `CUDA_HOME`

pytorch / tutorials

PyTorch tutorials.

https://pytorch.org/tutorials/

BSD 3-Clause "New" or "Revised" License

8.23k stars 4.06k forks source link

AOTI tutorial `CUDA_HOME` #3153

Open bhack opened 1 week ago

bhack commented 1 week ago

📚 The doc issue

I am trying to follow https://pytorch.org/tutorials/recipes/torch_export_aoti_python.html but it seems it doesn't work on our nightly runtime images cause missing CUDA_HOME.

Suggest a potential alternative/fix

No response

cc @svekars @brycebortree @sekyondaMeta @ezyang @chauhang @penguinwu @avikchaudhuri @gmagogsfm @zhxchen17 @tugsbayasgalan @angelayi @suo @ydwu4 @desertfire @chenyang78

bhack commented 1 week ago

/cc @agunapal @angelayi @svekars it is working only with devel image

agunapal commented 1 week ago

Hi @bhack I believe that is correct. Generation of the .so requires a devel image. That was my observation too when I tried it few months ago.

bhack commented 1 week ago

Is this what we want? If yes do we want to document it?

agunapal commented 1 week ago

Will update the tutorial once https://github.com/pytorch/pytorch/issues/140055 and https://github.com/pytorch/pytorch/issues/140053 are resolved. This likely already works in the nightlies