欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

win10安装pytorch

程序员文章站 2024-03-21 16:32:52
...

win10安装pytorch

首先确定自己电脑适用的版本

笔者电脑环境为:

  • python3.7
  • cuda9.2(可通过NVIDIA控制面板查询得到)

win10安装pytorch

win10安装pytorch

方法一:利用pip安装

pytorch官网中,根据自己的电脑配置选择适合的命令,打开cmd输入命令进行安装。

win10安装pytorch

方法二:使用conda安装

方法同上,不过选择Conda而不是Pip

由于在国内,上述两种方法同时无法成功

方法三:使用conda安装,但是通过清华镜像下载

我们需要在cmd中输入如下命令,将清华镜像添加到Anaconda的channel中

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

设置好之后,同样通过pytorch官网提供的命令进行安装:

在这里有个坑

官网提供的命令为:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

其中,-c pytorch参数指定了conda获取pytorch的channel,在此指定为conda自带的pytorch仓库。
因此,只需要将-c pytorch语句去掉,就可以使用清华镜像源快速安装pytorch了。

但是,笔者使用这种方法同样出错。

方法四:手动下载whl文件,进行安装

点这里中下载对应自己电脑配置的torch以及torchvision的whl文件

笔者下载的文件如下:

win10安装pytorch

win10安装pytorch

最后使用pip命令进行安装:

pip install torch-1.4.0+cu92-cp37-cp37m-win_amd64.whl
pip install torchvision-0.4.2+cu92-cp37-cp37m-win_amd64.whl

终于成功!