pytorch是学习神经网络常用的一个包,但是因为其2G的大小导致利用pip或者conda下载很容易出错(哪怕换了国内的镜像源),一旦网络超时就得推倒重来,我也在被折磨了很久以后发现了一个稳定可靠的方法,那就是抛弃在线安装,拥抱离线安装。
1、打开Anaconda prompt
2、创建一个虚拟环境
conda create -n name python=3.6
创建你的虚拟环境,name处就填你的虚拟环境的名字,因为我是用来安装pytorch,所以我就脚pytorch了。
3、进入虚拟环境
conda activate name
这里的name跟步骤2的name相同,就能进入我们刚刚创建的虚拟环境。
4、下载pyorch的安装包
进入torch的官网 https://pytorch.org/get-started/locally/
根绝红色方框里的的信息我们来下载离线安装包。
CUDA是通过显卡GPU来计算的,如果你有英伟达显卡的话可以打开显卡控制面板——帮助——系统信息——组件,来查看你的电脑的CUDA版本
如果没有显卡,选NONE就好了。
进入下面的网址下载离线包https://download.pytorch.org/whl/torch_stable.html
点击下载torch和torchvision包,也可以复制连接通过迅雷下载(实测更快)
4、安装离线包
打开Anaconda prompt并进入创建的虚拟环境
输入命令
pip install (后面接刚刚下载的包存放的完整路径)
我的如下
pip install D:\pytorch_dowload\torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
pip install D:\pytorch_dowload\torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl
5、测试安装是否成功
输入python然后导入torch,如果不报错,恭喜你成功了
6、最后要把虚拟环境导入jupyter方便我们编程,下面的就不赘述了。