网上很多Torch/Torchvision安装教程但是大部分都不实用,根据我的安装经验先介绍之前安装所出现的问题吧
Torch不像numpy/pandas等这些包体积小,在Prompt/cmd终端通过pip或conda就能快速安装,所以通过一行命令往往装到一半就超时了,即是通过清华、阿里源镜像安装还是会超时。
这篇Blog就是介绍通过源文件(xxx.whl)下载快速安装Torch的!
1、进入PyTorch官网,在首页选择安装版本
这里根据自己电脑的配置要求,选择操作系统、cpu/gpu等。一步安装的方法就是直接在Prompt中运行提示的指令就能够完成了,实际上torch源文件在国外,国内安装速度巨慢,即使有镜像源还是在漫长的等待中报错!这里看到上面我用红框标注的位置,就源文件所在URL。
2、打开源文件URL,搜索所需torch、torchvision安装文件
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同理!
安装完成后是这样的
5、测试是否安装成功
调出Prompt终端,在base环境中激活Python,打印torch版本,若打印成功则安装完成。
6、可选项
如果想要在其他虚拟环境中使用torch,只需将torch安装文件夹复制到自己所创建的虚拟环境的site-packages下即可。