windows平台PyTorch安装

一  PyTorch安装

1.使用pip安装

             目前,使用pip安装PyTorch二进制包是最简单,最不容易出错,同时也是最适合新手的安装方式。从PyTorch官网选择操作系统,包管理器pip,Python版本及CUDA版本,会对应不同的安装命令

            现在官网推出了window版本的PyTorch

复制官网的安装命令,打开cmd输入

pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-win_amd64.whl

可能在下载过程中出错,重新执行命令即可(我大概试了四次才成功下载)

安装好PyTorch后,还需安装Numpy,安装命令

全部安装后,打开python,运行如下命令

import  torch as t

没报错即表示PyTorch安装成功

 

2.使用conda安装

    conda是Anaconda自带的包管理器。如果使用Anaconda作为Python环境,则除了使用pip安装,还可以使用conda进行安装。同样,这次在官网选conda即可

打开Anaconda  Prompt 输入官网的命令

     conda的安装速度比较慢,建议国内用户,把conda源设置为清华tuna。在命令行输入如下指令即可

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes

即使是使用Anaconda的用户,也建议使用pip安装PyTorch,一劳永逸

 

二:遇到的问题

       前面说过,笔者用pip安装方式安装成功,在python环境(IDLE,或cmd)下可以成功导入torch包,但是在jupyter notebook中却提示no module named torch,再执行print(sys.path)指令,发现没有site packages的路径,笔者是这样解决的

       打开我的电脑属性->高级系统设置->环境变量->新增系统变量->

其中变量名为PYTHONPATH,值为C:\Users\chmusk\AppData\Local\Programs\Python\Python37;C:\Users\chmusk\AppData\Local\Programs\Python\Python37\Lib\site-packages 

注意两个路径之间使用英文分号分开,每个人python的安装位置不一样,要找Python37的路径可以直接搜索python.exe,然后打开文件所在位置,就行了。

现在往notebook里输入import torch as t 就不会报错了,如果报错,退出软件或者直接重启电脑就行了。

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

posted @ 2019-06-27 19:21  ch_musk  阅读(219)  评论(0编辑  收藏  举报