Windows快速安装Torch1.8.1+torchvision0.9最新教程

网上很多Torch/Torchvision安装教程但是大部分都不实用,根据我的安装经验先介绍之前安装所出现的问题吧
Torch不像numpy/pandas等这些包体积小,在Prompt/cmd终端通过pip或conda就能快速安装,所以通过一行命令往往装到一半就超时了,即是通过清华、阿里源镜像安装还是会超时。

这篇Blog就是介绍通过源文件(xxx.whl)下载快速安装Torch的!
1、进入PyTorch官网,在首页选择安装版本
Windows快速安装Torch1.8.1+torchvision0.9最新教程_第1张图片
这里根据自己电脑的配置要求,选择操作系统、cpu/gpu等。一步安装的方法就是直接在Prompt中运行提示的指令就能够完成了,实际上torch源文件在国外,国内安装速度巨慢,即使有镜像源还是在漫长的等待中报错!这里看到上面我用红框标注的位置,就源文件所在URL。

2、打开源文件URL,搜索所需torch、torchvision安装文件
Windows快速安装Torch1.8.1+torchvision0.9最新教程_第2张图片
torchvision检索.whl安装文件同理,然后使用迅雷快速下载。注意:cu表示CUDA版本,cp表示Python版本,一定要对应不然会报错C extensions!如下:

ImportError: Failed to load PyTorch C extensions:
    It appears that PyTorch has loaded the `torch/_C` folder
    of the PyTorch repository rather than the C extensions which
    are expected in the `torch._C` namespace. This can occur when
    using the `install` workflow. e.g.
       ......

下载完成后如下所示:
在这里插入图片描述
3、将下载文件放到Anaconda的Script目录下
我的目录是 D:\Program\Anaconda\Scripts,根据自己下载的Anaconda位置选择。

4、在Script目录下(也就是torch目标文件位置)调出PowerShell,运行easy_install.exe程序,按照格式为pip install xxxxx.whl进行安装,
我的是pip install torch-1.8.0+cu111-cp38-cp38-win_amd64.whl,回车就开始安装了,torchvision同理!
安装完成后是这样的
Windows快速安装Torch1.8.1+torchvision0.9最新教程_第3张图片
5、测试是否安装成功
调出Prompt终端,在base环境中激活Python,打印torch版本,若打印成功则安装完成。
Windows快速安装Torch1.8.1+torchvision0.9最新教程_第4张图片
6、可选项
如果想要在其他虚拟环境中使用torch,只需将torch安装文件夹复制到自己所创建的虚拟环境的site-packages下即可。
Windows快速安装Torch1.8.1+torchvision0.9最新教程_第5张图片

你可能感兴趣的:(Pytorch)