Windows环境下安装PyTorch,并验证PyTorch是否安装成功

PyTorch是一个由Facebook的人工智能研究团队开发的开源深度学习框架,它是用Python语言编写,是Python的一种扩展,这使得其更易于学习和使用,下面将介绍在Windows环境下安装pytorch并验证是否安装成功。

检查系统环境

在开始安装PyTorch之前,请确保你的系统满足下列要求:

Python版本:3.8或更高

我用的是conda,默认的python版本是3.11
在这里插入图片描述
注意:如果用的是conda去管理你的包,需要切换到对应的环境,再去下载包,否则会存在包找不到的问题
使用下面的命令进入对应的环境

conda activate conda

成功后在命令行前面会有该环境的名称
在这里插入图片描述

访问pytorch官网(https://pytorch.org/)

浏览器输入上面网址访问官网主页,选择合适的版本进行安装,安装命令官网会提供
在这里插入图片描述
我这边用的是conda,所以用的conda命令,如果没有装Acondana,可以使用pip命令,但注意版本不能过低
在这里插入图片描述
执行脚本后看到done,代表已经安装成功了(注:我是在自己命名的conda环境中安装的,切记要切到对应环境),使用conda list查看安装的包,pip安装的使用pip list查看
在这里插入图片描述

验证

验证的话我们可以直接在命令行验证,使用python 命令进去,复制下面指令到命令行运行

import torch

print(torch.__version__)  # pytorch版本
print(torch.version.cuda)  # cuda版本
print(torch.cuda.is_available())  # 查看cuda是否可用

运行结果如图
在这里插入图片描述
还可以再pycharm工具中验证,新建一个python文件,将上面的命令复制到文件中,运行如下图:
在这里插入图片描述
验证成功,至此,pytorch安装跟验证已完结了,希望对读者朋友有帮助

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值