安装pytorch时,文件名称的识别

这里记录一下在安装pytorch时,下载whl文件时,选择版本的问题。
首先提一下pytorch安装。
有显卡的话,是需要先安装cuda的,cuda的安装是需要根据你自己电脑的显卡型号来安装,具体可百度。然后根据cuda型号和自己电脑的系统和python版本来下载安装pytorch。
没有显卡的话,就只能安装cpu版本的pytorch。这就只需要根据自己电脑的系统和python的版本来下载安装pytorch。
首先到官网https://pytorch.org/,往下面拉,可以看到让你选择配置。
你可以在选择配置后,根据Run this Command里自动生成的代码,在命令行里进行下来。但是一般情况下,都非常慢或者出现很多问题,即使是换了国内的源。
在这里插入图片描述
所以建议直接下载whl文件进行安装。
网址:https://download.pytorch.org/whl/torch_stable.html
在这个网站下载所需要版本的torch和torchvision,然后再命令行里pip install 进行安装就行了。
下面说明这些文件名称的含义。
在这里插入图片描述
最前面的cpu就是cpu版本的意思,如果要gpu版本的,前面就是cu开头,cu后面的数字就是cuda的版本号,你自己用的什么版本的cuda就下载哪个版本的。
查询cuda版本ÿ

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值