I don’t want to hijack the thread, but at least the title fits.
The case here is that the home fold where my script lives is NFS mounted from another server. When comes to model.cuda(), pytorch takes some time to move something over the NFS link.
But only my script is on NFS. Both pytorch under conda and the data is on local disk.
So I guess when pytorch compile those cuda source file it uses the same directory where the script lives in or user’s home directory?
(because I run the script in local directory, model.cuda() still takes some time, thus pytorch don’t use working directory?)
The question is is there a way(or simple modification) to ask pytorch to use a local path to do compiling work if all the hypothesis is correct?