Great news! PyTorch now is supporting Windows!
If you have a PC with suitable Nvidia graphics card and installed CUDA 9.0 and Anaconda, type the following commands;
conda install pytorch cuda90 -c pytorch
pip3 install torchvision
It is about 500 MB, so be patient!
Underline is the old post.
PyTorch is a deep learning framework that puts Python first. Currently, it only supports MacOS or Linux.
But, can we use it on WIN10 without changing the system/computer?
Yes, we can.
The following source is from the Chinese ZhiHu forum PyTorch在64位Windows下的Conda包.
To install PyTorch, you will need the following requirements:
- Anaconda3 x64 (with Python 3.5/3.6)
- Windows 64 （Windows 7 or Windows Server 2008 above）
- GPU needs CUDA Packages (including the main bin files of CUDA 8 / 9)
Now, you can install the PyTorch by the following commands, choose the one that suits your PC.
# for CPU only packages conda install -c peterjc123 pytorch-cpu # for Windows 10 and Windows Server 2016, CUDA 8 conda install -c peterjc123 pytorch # for Windows 10 and Windows Server 2016, CUDA 9 conda install -c peterjc123 pytorch cuda90 # for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8 conda install -c peterjc123 pytorch_legacy
The following is my result:
Use the following codes to test “cuda” and “cudnn”:
# CUDA TEST
x = torch.Tensor([1.0])
xx = x.cuda()
# CUDNN TEST
from torch.backends import cudnn