pytorch安装实录(win10+cuda8+pycharm+anaconda)

由于TensorFlow的静态图特点导致最近的一些实验无法完成,决定转战pytorch,自己做一个安装记录,也希望对有需要的小伙伴有所帮助

先前安装的是cuda8.0,并非最新版本,因此需要安装对应版本的pytorch。在pytorch官网上https://pytorch.org/可以找到最新版本以及之前的版本

在cmd中直接安装即可

有的小伙伴(包括我)会发现直接使用pip安装基本不动,使用镜像也不动(镜像可参考https://blog.csdn.net/xo3ylAF9kGs/article/details/104104041/

这里有一个比较好的解决方法是直接下载轮子,cuda8.0+python3.6可以使用https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl,更多版本的下载地址可参考https://blog.csdn.net/qq_27825451/article/details/90033909

更多轮子https://download.pytorch.org/whl/cu100/torch_stable.html

下载到本地后再安装简直不要太快

使用pip install <本地下载地址>即可

接下来是测试是否安装成功的环节

import torch

以及

import torchvision

在这步中,可捕获错误

ImportError: No module named torchvision

这里通过查找资料可以发现与cuda8.0+python3.6+pytorch1.0对应的torchvision版本为0.4

然后安装

pip install torchvision

可以得到如下四种错误

ERROR: Could not find a version that satisfies the requirement torch==1.4.0 (from torchvision) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2)

ERROR: No matching distribution found for torch==1.4.0 (from torchvision)

ERROR: Could not find a version that satisfies the requirement torchvision==0.4 (from versions: 0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.2.0, 0.2.1, 0.2.2, 0.2.2.post2, 0.2.2.post3, 0.3.0, 0.4.1, 0.5.0)

ERROR: No matching distribution found for torchvision==0.4

可以在提示信息中发现有0.4.1这个版本,再次尝试

pip3 install torchvision==0.4.1

这里由于某种未知力量,使用了pip3

再次测试

好的,开启pytorch大门!

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值