python安装pytorch-cpu(Window系统64bit)

【时间】2019.08.15

【题目】python安装pytorch-cpu(Window系统64bit)

参考官网:https://pytorch.org/get-started/locally/

需要安装torch和torchvision库

一、使用anaconda安装:

conda install pytorch-cpu torchvision-cpu -c pytorch

下载太慢的话直接下载whl文件再安装:http://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl

修改torch-1.1.0-cp36-cp36可以直接获取对应版本。

本人使用的是python3.6.5,目前的最新版本是torch-1.1.0-cp36-cp36。

然后直接cd到下载目录,再安装:

conda install --use-local xxx.whl

但本人安装时出错,最后只能在anaconda prompt中使用pip安装:

pip install xxx.whl

二、使用pip安装

# Python 2.7
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27m-linux_x86_64.whl
# Python 3.5
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
# Python 3.6
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
# Python 3.7
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

 三、简单测试安装是否成功

  • python
  • import torch
  • print(torch.__version__)

成功会输出版本号。

 

你可能感兴趣的:(python库)