Windows + CUDA working with PyTorch!


(Devin G Bost) #1

To anyone running Windows and wanting to be able to build their models on their Windows machine, I was able to get PyTorch working on Windows 10 x64 with CUDA 7.5. Others have reported getting CUDA 8.0 working on Windows 10 x64. See: PyTorch Windows Support for more information.

(Devin G Bost) #2

As of three hours ago, the owner of the PyTorch library at Facebook AI research mentioned that they are planning on adding Windows support in the next version of PyTorch, v0.4. See