验证pytorch是否安装成功

Anaconda操作——验证pytorch安装

提示:下面展示的是如何在anaconda的命令行验证pytorch安装是否安装成功,并且查看是否成功使用自己电脑的GPU。


文章目录

  • Anaconda操作——验证pytorch安装
  • 1. 激活对应的虚拟环境
  • 2. 查看pytorch包
  • 3. 进入虚拟环境的python环境
  • 4. 调用Pytorch工具包
  • 5. 接着输入命令 torch.cuda.is_ available()
  • 总结


1. 激活对应的虚拟环境

首先在开始菜单栏,打开conda的命令行,其中带有PS的版本是power shell专业版本的命令行,另一个为普通版本,本质没有区别,用哪一个都可以。
验证pytorch是否安装成功_第1张图片

进入命令提示符,输入命令conda activate 虚拟环境名,进入我们创建的虚拟环境。
验证pytorch是否安装成功_第2张图片

2. 查看pytorch包

注意:进入自己创建的虚拟环境之后,可以使用命令conda list查看一下自己的虚拟环境的pytorch的这些包,看有没有pytorch或者torch。

如下图:

验证pytorch是否安装成功_第3张图片

注意:可以看到设备的pytorch版本和cuda版本等信息。

3. 进入虚拟环境的python环境

进入到python环境:
在这里插入图片描述

4. 调用Pytorch工具包

输入import torch命令就可以调用pytorch工具包,注意:这个命令的运行可能需要时间,需要稍微等一下,才能出现三个箭头。

5. 接着输入命令 torch.cuda.is_ available()

如果显示True,就说明我们这个PyTorch安装成功了,说明我们安装的是GPU版本,我们的pytorch是可以使用我们的GPU的。
验证结果如下:

在这里插入图片描述

总结

以上就是一个简单的Anaconda操作,验证pytorch是否安装成功的一个具体的操作,大家可以下去试一试,比较基础,大神勿喷,欢迎交流。谢谢!

你可能感兴趣的:(深度学习,pytorch,人工智能,python,深度学习,机器学习,conda)