Shakerato

Install PyTorch using Anaconda on Windows 10 본문

Research

Install PyTorch using Anaconda on Windows 10

Shakeratto 2018. 6. 10. 19:50

1. check cuda9.0 installed (nvcc --version)

   (if yet, install cuda) https://developer.nvidia.com/cuda-90-download-archive

   (if no gpu) pass this step


2, Install Anaconda

   https://www.anaconda.com/download/


3. in 'cmd'

   conda create -n pytorch python=3.6 anaconda


4. in 'cmd'
   conda install pytorch cuda90 -c pytorch 

   (if no gpu) conda install pytorch-cpu -c pytorch


5. in 'cmd'

   pip3 install torchvision

If you got a error message like this,

  ' RuntimeError: PyTorch does not currently provide packages for PyPI' 

then install torchvision using follow command


pip install torchvision


6. Try example codes

   https://github.com/pytorch/examples


Comments