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 就不会报错了,如果报错,退出软件或者直接重启电脑就行了。