Anaconda安装cpu版PyTorch框架(Windows)

记录贴,面向小白。方案一使用清华源下载,方案二使用whl文件下载。推荐直接看方案二,可以安装指定版本,下载速度也很快。

方案一

首先,打开Anaconda Prompt,
输入conda env listconda info -e, 查看当前所有的隔离环境。
在这里插入图片描述
找到你想安装pytorch的环境,激活,我这里选择的是Python3.7,所以activate py37
在这里插入图片描述
这时,前面括号里的base就切换到了py37,说明环境已激活成功。
若你想新建一个环境来安装pytorch,那么输入

conda create --name {env_name} python={python_version}
例: conda create --name py37 python=3.7

这样新环境就创建成功啦。接下来进入正题。

pytorch官网看一下,
在这里插入图片描述
若是直接输入上述官网给的命令,估计速度太慢,还有可能会连接中断。所以,我这里采用的是conda清华源镜像

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 --set show_channel_urls yes

然后,输入

conda install pytorch-cpu torchvision-cpu -c pytorch

有博主说,把最后的-c pytorch去掉会下的快一点,但我是没去下得更快,大家可以自己试一下。
然后就直接安装,也没什么了。

中间出了两次错误:

Error 1
CondaError: Downloaded bytes did not match Content-Length
  url: https://repo.anaconda.com/pkgs/main/linux-64/cudnn-7.0.5-cuda8.0_0.tar.bz2
  target_path: /home/yyf/miniconda3/pkgs/cudnn-7.0.5-cuda8.0_0.tar.bz2
  Content-Length: 261398285
  downloaded bytes: 47463195

在用conda安装包的时候下载包的长度不够导致安装包不成功。原因一般是在下载的时候速度较慢,导致下载timeout而终止。对,用清华源镜像也出现了这个问题,简单一点的办法就是,设置conda下载的timeout,然后再尝试。

conda config --set remote_read_timeout_secs 1000.0

若是不想这样的,可以有其他的方式

Error 2
UserWarning: mkl-service package failed to import, therefore Intel(R) MKL initialization ensuring it

先去自己环境里看一下,发现是有mkl-service,但是没导进来,这个是咱的锅。我之前直接安装的是python编译器,后来才安的Anaconda. 结果系统环境变量没有改过来。

  • 更改环境变量
    右击此电脑,选择属性(或通过控制面板,进入‘控制面板\系统和安全\系统’)
    在这里插入图片描述
    进入高级系统设置,点击环境变量,在系统变量中找到Path,编辑,在其中添加
D:\Anaconda3\Library\bin
D:\Anaconda3\Scripts
D:\Anaconda3

我的Anaconda包是放在了D盘,大家照这个格式把自己anaconda的位置写进去就行了。
修改后,一路保存,否则可能修改不成功。然后anaconda的环境变量就配好啦。

方案二

torch和torchvision版本对应关系

torchtorchvisionpython
1.5.10.6.1>=3.6
1.5.00.6.0>=3.6
1.4.00.5.0==2.7, >=3.5, <=3.8
1.3.10.4.2==2.7, >=3.5, <=3.7
1.3.00.4.1==2.7, >=3.5, <=3.7
1.2.00.4.0==2.7, >=3.5, <=3.7
1.1.00.3.0==2.7, >=3.5, <=3.7
<1.0.10.2.2==2.7, >=3.5, <=3.7

https://download.pytorch.org/whl/torch_stable.html
这种方式真的超快,超方便。
这个方法是我在重装pytorch时找到的。因为第一次装的是1.1版本,太老了,没有Transformer模型。只能卸了重安。1.2版本就有Transformer模型了。

以torch-1.3示例,先到上面那个网站把你需要的torch和torchvision轮子下载下来。
cp37就是python3.7的意思,后面跟着的是系统类型,以此类推,选择自己需要的便好。

进入anaconda,激活你要安装的那个环境,然后改变路径到你放置torch和torchvision的地方,我放在了d盘
在这里插入图片描述

输入

pip install torch-1.3.0+cpu-cp37-cp37m-win_amd64.whl
pip install torchvision-0.4.1+cpu-cp37-cp37m-win_amd64.whl

等待安装成功就行了。

测试

在这里插入图片描述
导入没问题,就是安装成功啦。


  • 1
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值