Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pytorch on windows does not support multiprocessing #6

Open
powerspowers opened this issue May 22, 2019 · 6 comments
Open

pytorch on windows does not support multiprocessing #6

powerspowers opened this issue May 22, 2019 · 6 comments

Comments

@powerspowers
Copy link

I'm attempting to figure out which bits of the code attempt to spawn new processes since this fails on Windows 10 pytorch. I had to modify pcgan in a similar manner to get it to work recently.

@powerspowers
Copy link
Author

Wrapping the main calls at the bottom of pggan.py in main seems to have solved the multiprocessing problem

if __name__ == '__main__':

@powerspowers
Copy link
Author

Also appears you have to remove Volatile=true since that's been deprecated

@powerspowers
Copy link
Author

In addition all data[0] references need to be switched to item() which is a result of the Volatile tag being deprecated

https://github.com/pytorch/pytorch/issues/6061

@powerspowers
Copy link
Author

The last thing I had to do to get this working on Windows 10 pytorch was change all the directory and file creation paths to use backslashes instead of forward slashes. I also have to remove the '-p' flag in mkdir calls. I commented out the tensorboard code section until I can spend some time installing it. So far it seems to be running well on my Dell XPS with an nVidia 1080 (6GB).

@powerspowers
Copy link
Author

Also had to add in the following modification because DataParallel puts the original object into the module variable

self.G.module.grow_network(floor(self.resl)) self.D.module.grow_network(floor(self.resl))

@powerspowers
Copy link
Author

In intermediate_block you have to cast the ndim as an int because in python3 dividing two integers results in a float. grrr only collided into this one when the resl got to 6 and the code to divide ndim by two clicked in.

ndim = int(ndim / 2)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant