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安装跟验证已完结了,希望对读者朋友有帮助