Pergunta

I was wondering, if there is any way to install pytorch in windows like the way we can install tensorflow. My machine is not supporting docker.

Foi útil?

Solução

PyTorch should work fine in WSL (CPU only).

Outras dicas

As of August 14, 2017, you can install Pytorch from peterjc123's fork as follows. Currently, python 3.5 and 3.6 are supported.

# If your main Python version is not 3.5 or 3.6
conda create -n test python=3.6 numpy pyyaml mkl

# for CPU only packages
conda install -c peterjc123 pytorch

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80

# 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 cuda80

pytorch support for windows Link

Licenciado em: CC-BY-SA com atribuição
scroll top