@smy20011 This doesn't work for cuda 11.7/11.8(switching the image for the appropriate pytorch version doesn't work)
@smy20011
This doesn't work for cuda 11.7/11.8(switching the image for the appropriate pytorch version doesn't work)