pytorch的安装(GPU版本和CPU版本都可以)

pytorch的安装(GPU版本和CPU版本都可以)

直接去download.pytorch.org/whl/torch_stable.html该网站下载torch,如下图所示:

image-20231212161353274

其中,前面的cu表示cuda,118表示cuda的版本号为11.8,torch-2.1.0表示torch的版本号为2.1.1,cp38为python版本为3.8,310表示python版本为3.10,win、linux代表不同的系统。

cpu版本

注意安装时候看自己的电脑有没有GPU,如果没有的话安装gpu版本应该会出现问题,要安装cpu版本。但安装cpu版本建议可以直接

pip install torch -i https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

安装会更加快捷。

gpu版本

若不知道自己的cuda版本号,windows系统可以打开命令行输入nvidia-smi,即可查看,如下所示:

image-20231212161610429

python的版本号可以直接python -v进行查看

知道了自己电脑cuda和python的版本号之后,就可以进入上面所写的链接中,进行下载。

注意:上文中讲到的版本号要对应好,不然很容易出现错误!!!

下载完成之后,就可以使用pip instasll 的方式来进行安装whl文件,若安装的环境为虚拟环境的话,记得要激活自己的虚拟环境,然后在虚拟环境下面进行

pip install 文件名

例:pip install torch-1.9.0+cu111-cp39-cp30-win_amd.64.whl

注意(我这个写的默认已经在此文件的安装目录了,如果不在,要么选择cd到这个目录,要么在此文件的前面加上路径名如”pip install e://test/torch-1.9.0+cu111-cp39-cp30-win_amd.64.whl“)

注意

下面写的这个方法,我没有亲自试,建议大家还是直接装自己版本所对应的,实在找不到再采用下面的这种方式。

若出现

whl is not a supported wheel on this platform

则可以pip debug --verbose命令查看平台支持的版本

image-20231212162223980

对比我们的文件名,发现不在这个支持的版本列表里面,很简单,修改whl文件名即可,我修改后的文件名如下:torch-1.12.1+cu116-cp39-abi3-win_amd64.whl,其实对照上图中的任何一个的名称都可以,然后再用pip install torch-1.12.1+cu116-cp39-abi3-win_amd64.whl 安装,成功。

安装成功之后可以

import torch
print(torch.cuda.is_available() )

若输出为true则代表torch的gpu版本就安装成功了,若为false,表示gpu不可用。

参考链接:

若输出为true则代表torch的gpu版本就安装成功了,若为false,表示gpu不可用。

参考链接:

用whl安装pytorch踩坑实录(whl is not a supported wheel on this platform.)_安装pytorch失败-CSDN博客

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值